ray/ci/regression_test/rllib_stress_tests/wait_cluster.py
2020-02-24 21:18:53 -08: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)