diff --git a/deploy_spherex_qc.py b/deploy_spherex_qc.py index 8c7f58eaf1716918b92932a4350b643198c38b6e..5f5e26d40e247eeee891facfcab626c2e19972aa 100644 --- a/deploy_spherex_qc.py +++ b/deploy_spherex_qc.py @@ -22,7 +22,10 @@ if not os.path.exists(fn): print("file downloaded") -ray.init("auto") +runtime_env = {"working_dir": ".", "excludes": ["data", "__pycache__"]} +print(runtime_env) + +ray.init("auto", runtime_env=runtime_env) # Load the data and put it on the shared memory fn = "data/zodicube2_4x6_21Mar.npy" @@ -59,8 +62,14 @@ releases_info["releases"].append(dict(release_name="kasi20211021pre01", release_date="2021-10-21", api_prefix=api_prefix)) +ray_actor_options = dict() + +n_spherex_resources = len([node for node in ray.nodes() if "spherex" in node["Resources"]]) +if n_spherex_resources: + ray_actor_options["resources"] = dict(spherex=1) +print(ray_actor_options) -@serve.deployment(route_prefix=api_prefix) +@serve.deployment(route_prefix=api_prefix, ray_actor_options=ray_actor_options) class ReleaseHandler: def __init__(self, qc_handle): self.qc_handle = qc_handle