ray/release/rllib_tests/stress_tests/wait_cluster.py
2020-10-22 17:04:41 -07:00

10 lines
219 B
Python

import ray
import time
ray.init(address="auto")
curr_nodes = 0
while not curr_nodes > 8:
print("Waiting for more nodes to come up: {}/{}".format(curr_nodes, 8))
curr_nodes = len(ray.nodes())
time.sleep(5)