Function build_client

espnet3/parallel/parallel.py:171–189

Create or retrieve a Dask client using the provided or global configuration. Args: config (DictConfig, optional): Cluster config. If None, uses global one. Returns: Client: Dask client instance.

(config: DictConfig = None)

def build_client(config: DictConfig = None) -> Client:
    """Create or retrieve a Dask client using the provided or global configuration.

    Args:
        config (DictConfig, optional): Cluster config. If None, uses global one.

    Returns:
        Client: Dask client instance.
    """
    if config is not None:
        set_parallel(config)
        return _build_client(config)

    if parallel_config is None:
        raise ValueError(
            "Parallel configuration not set. Use `set_parallel` to set it."
        )

    return _build_client(parallel_config)


class DictReturnWorkerPlugin(WorkerPlugin):