diff --git a/backend/src/app/preprocessing/ray_model_worker/models/cota.py b/backend/src/app/preprocessing/ray_model_worker/models/cota.py index a6ecdb4a6..257bd624e 100644 --- a/backend/src/app/preprocessing/ray_model_worker/models/cota.py +++ b/backend/src/app/preprocessing/ray_model_worker/models/cota.py @@ -19,7 +19,7 @@ SHARED_REPO_ROOT: Path = Path(conf.repo_root) -LOGGER = logging.getLogger("ray.serve") +logger = logging.getLogger("ray.serve") cota_conf: Dict = build_ray_model_deployment_config("cota") @@ -92,7 +92,7 @@ def __finetune(self, input: RayCOTAJobInput) -> Tuple[SetFitModel, List[str]]: ) # 2. load a SetFit model from Hub - LOGGER.info(f"Loading COTA model {MODEL} on {DEVICE}") + logger.info(f"Loading COTA model {MODEL} on {DEVICE}") model = SetFitModel.from_pretrained(MODEL, device=DEVICE) # 3. init training