chore: move @ for model alias outside alias_name

pass-culture · Sep 17, 2024 · c0a7993 · c0a7993
1 parent 6667c4a
commit c0a7993
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apps/fraud/compliance/api/src/pcpapillon/utils/model_handler.py b/apps/fraud/compliance/api/src/pcpapillon/utils/model_handler.py
@@ -22,7 +22,7 @@ class ModelWithMetadata:
 
 
 class ModelHandler:
-    MODEL_ALIAS = "@production"
+    MODEL_ALIAS = "production"
 
     def __init__(self) -> None:
         custom_logger.info("Connecting to mlflow")
@@ -57,4 +57,4 @@ def get_model_hash_from_mlflow(self, model_name: str):
 
     @staticmethod
     def _get_mlflow_model_name(model_name: ModelName):
-        return f"api_{model_name.value}_{ENV_SHORT_NAME}{ModelHandler.MODEL_ALIAS}"
+        return f"api_{model_name.value}_{ENV_SHORT_NAME}@{ModelHandler.MODEL_ALIAS}"