Class DaskWorkerConfig

benchmarks/src/environment/dask.py:27–40

One Dask worker corresponds to one node.

@dataclasses.dataclass(frozen=True)
class DaskWorkerConfig:
    """
    One Dask worker corresponds to one node.
    """

    processes: int
    threads_per_process: int
    node: Optional[int] = None
    init_cmd: List[str] = dataclasses.field(default_factory=list)

    @staticmethod
    def create(count: int, n_processes: int = 1, **kwargs) -> "DaskWorkerConfig":
        assert count % n_processes == 0
        return DaskWorkerConfig(processes=n_processes, threads_per_process=count // n_processes, **kwargs)


@dataclasses.dataclass(frozen=True)