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)