Method setup

espnet3/parallel/parallel.py:203–211

Set up the worker by calling the user-defined setup function.

(self, worker)

        self.setup_fn = setup_fn

    def setup(self, worker):
        """Set up the worker by calling the user-defined setup function."""
        env = self.setup_fn()
        if not isinstance(env, dict):
            raise ValueError("setup_fn must return a dict")
        worker.plugins["env"] = env

        # Set worker id so that users can use it for identifing workers
        os.environ["DASK_WORKER_ID"] = str(worker.id)


def wrap_func_with_worker_env(func: Callable) -> Callable: