espnet3.parallel.base_runner.get_job_cls

source

espnet3.parallel.base_runner.get_job_cls(cluster, spec_path=None)

Create a Dask Job class that runs an async runner spec file.

This wraps the cluster’s job_cls and injects a command template that executes this module as a script with the given spec file path.

• Parameters:
  • cluster – Dask cluster instance that provides a job_cls attribute.
  • spec_path – Path to a JSON spec file consumed by the worker entrypoint.
• Returns: A dynamically created job class that submits async runner jobs.
• Return type: type
• Raises:AssertionError – If spec_path is not provided.

Example

>>> job_cls = get_job_cls(cluster, spec_path="spec.json")