utilities

class pyrosetta.distributed.cluster.utilities.SchedulerManager

Bases: object

Dask utility manager for PyRosettaCluster.

_setup_clients_dict() → Dict[int, Client]

Setup Dask clients dictionary for PyRosettaCluster.

_get_cluster() → Union[LocalCluster, SGECluster, SLURMCluster]

Given user input argument values, return the requested Dask cluster instance.

_setup_clients_cluster_adaptive() → Tuple[Dict[int, Client], Optional[Union[LocalCluster, SGECluster, SLURMCluster]], Optional[Adaptive]]

Given user input arguments, return the requested Dask client, cluster, and adaptive instance.

_maybe_adapt(adaptive: Optional[Adaptive]) → None

Adjust the maximum number of Dask workers.

_maybe_teardown(clients: Dict[int, Client], cluster: Optional[Union[LocalCluster, SGECluster, SLURMCluster]]) → None

Teardown the Dask client and cluster.

pyrosetta.distributed.cluster.utilities.get_dask_worker() → Worker

Get the current Dask worker.