lowercase logger

uhh-lt · Jun 12, 2024 · 4e8911e · 4e8911e
1 parent 5218ef6
commit 4e8911e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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