diff --git a/deploy_spherex_qc.py b/deploy_spherex_qc.py index 127943353f911a902d26bd635cb44cfa37acae42..8c7f58eaf1716918b92932a4350b643198c38b6e 100644 --- a/deploy_spherex_qc.py +++ b/deploy_spherex_qc.py @@ -8,8 +8,6 @@ from ray import serve import os import numpy as np -from spherex_qc import SPHERExQCHandler as _SPHERExQCHandler - # download data minio_endpoint = "http://210.110.233.67:9005" bucket = "public" @@ -33,15 +31,18 @@ zodi_data = dict((int(k), data[k]) for k in "123456") r_zodi_data = ray.put(zodi_data) - -class SPHERExQCHandler(_SPHERExQCHandler): +@serve.deployment +class SPHERExQCDeployment: "Derived class to suuport ray.get for data loading" def __init__(self, r_zodi_data): + from spherex_qc import SPHERExQCHandler as _SPHERExQCHandler zodi_data = ray.get(r_zodi_data) - _SPHERExQCHandler.__init__(self, zodi_data) + self.handler = _SPHERExQCHandler(zodi_data) + def query(self, name, ra, dec, zodi=False): + return self.handler.query(name, ra, dec, zodi) -SPHERExQCDeployment = serve.deployment(SPHERExQCHandler) +#SPHERExQCDeployment = serve.deployment(SPHERExQCHandler) SPHERExQCDeployment.deploy(r_zodi_data)