ablator.mp package#

Submodules#

ablator.mp.node_manager module#

class ablator.mp.node_manager.NodeManager(private_key_home: Path, ray_address: str | None = None)[source]#

Bases: object

available_resources(node_ips: list | str | None = None, timeout: int | None = 60) → dict[str, ablator.mp.node_manager.Resource][source]#
property node_ips: list[str]#
run_cmd(cmd, node_ips: list | str | None = None, timeout: int = 60) → dict[str, str][source]#
run_lambda(fn, node_ips: list | str | None = None, timeout: int | None = 60)[source]#
update(timeout: int | None = 10)[source]#
utilization(node_ips: list | str | None = None, timeout: int | None = 60) → dict[str, ablator.mp.node_manager.Resource][source]#
class ablator.mp.node_manager.Resource(gpu_free_mem: dict[str, int], mem: int, cpu_usage: float, cpu_count: int, running_tasks: list[str] = <factory>)[source]#

Bases: object

cpu_count: int#
property cpu_mean_util: float#
cpu_usage: float#
gpu_free_mem: dict[str, int]#
property gpu_free_mem_arr: ndarray#
property least_used_gpu#
mem: int#
running_tasks: list[str]#
ablator.mp.node_manager.make_private_key(home_path: Path)[source]#
ablator.mp.node_manager.utilization()[source]#

ablator.mp.utils module#

Module contents#