diff --git a/deploy_spherex_qc.py b/deploy_spherex_qc.py index 5f5e26d40e247eeee891facfcab626c2e19972aa..494e983616944a7a98dcd20fcd7231142c8d6c59 100644 --- a/deploy_spherex_qc.py +++ b/deploy_spherex_qc.py @@ -34,7 +34,15 @@ zodi_data = dict((int(k), data[k]) for k in "123456") r_zodi_data = ray.put(zodi_data) -@serve.deployment +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(ray_actor_options=ray_actor_options) class SPHERExQCDeployment: "Derived class to suuport ray.get for data loading" def __init__(self, r_zodi_data): @@ -62,14 +70,7 @@ 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, ray_actor_options=ray_actor_options) +@serve.deployment(route_prefix=api_prefix) class ReleaseHandler: def __init__(self, qc_handle): self.qc_handle = qc_handle