trinity.utils.dlc_utils module#

class trinity.utils.dlc_utils.ClusterStatus[source]#

Bases: object

__init__()[source]#
finish() None[source]#
running() bool[source]#
trinity.utils.dlc_utils.get_dlc_env_vars() dict[source]#
trinity.utils.dlc_utils.is_running() bool[source]#

Check if ray cluster is running.

trinity.utils.dlc_utils.wait_for_ray_setup() None[source]#
trinity.utils.dlc_utils.wait_for_ray_worker_nodes(world_size: int) None[source]#
trinity.utils.dlc_utils.setup_ray_cluster(namespace: str) str[source]#

Setup a ray cluster in DLC environment.

This function will start a ray cluster if it is not running, otherwise it will reuse the existing ray cluster.

Returns:

The address of the ray cluster.

Return type:

str

trinity.utils.dlc_utils.stop_ray_cluster(namespace: str)[source]#

Stop the ray cluster by sending a signal to the cluster status actor.