wehub-resource-sync
33 lines
794 B
Python
33 lines
794 B
Python
import time
|
|
|
|
import ray
|
|
|
|
|
|
def main():
|
|
"""Submits custom resource request.
|
|
|
|
Also, validates runtime env data submitted with the Ray Job that executes
|
|
this script.
|
|
"""
|
|
|
|
# Workers and head are annotated as having 5 "Custom2" capacity each,
|
|
# so this should trigger upscaling of two workers.
|
|
# (One of the bundles will be "placed" on the head.)
|
|
ray.autoscaler.sdk.request_resources(
|
|
bundles=[{"Custom2": 3}, {"Custom2": 3}, {"Custom2": 3}]
|
|
)
|
|
|
|
while (
|
|
ray.cluster_resources().get("Custom2", 0) < 3
|
|
and ray.cluster_resources().get("Custom2", 0) < 6
|
|
):
|
|
time.sleep(0.1)
|
|
|
|
# Output something to validate the job logs.
|
|
print("Submitted custom scale request!")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
ray.init("auto")
|
|
main()
|