ray/release/nightly_tests/placement_group_tests/cluster.py
SangBin Cho 22f4ffed08
Disable cpu-only-nodes preferred scheduling that breaks placement groups. (#19129)
* Add a regression test for the short term

* done

* address code review

* lint
2021-10-07 05:34:04 -07:00

13 lines
256 B
Python

import time
from ray.cluster_utils import Cluster
cluster = Cluster()
cluster.add_node(num_cpus=16)
time.sleep(20)
print("Scaling up.")
cluster.add_node(num_cpus=16, num_gpus=1)
print("Scaled up. Waiting for 1000 seconds until done.")
time.sleep(1000)