fix: change model names

pass-culture · Sep 17, 2024 · 59d7e19 · 59d7e19
1 parent 17b090f
commit 59d7e19
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/apps/fraud/compliance/api/src/pcpapillon/utils/constants.py b/apps/fraud/compliance/api/src/pcpapillon/utils/constants.py
@@ -6,5 +6,5 @@ class ModelName(Enum):
     Enum class for model names
     """
 
-    OFFER_CATEGORISATION = "offer_categorisation"
-    COMPLIANCE = "compliance"
+    OFFER_CATEGORISATION = "offer_categorization"
+    COMPLIANCE = "compliance_default"
diff --git a/apps/fraud/compliance/api/src/pcpapillon/utils/model_handler.py b/apps/fraud/compliance/api/src/pcpapillon/utils/model_handler.py
@@ -57,6 +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}_default_{ENV_SHORT_NAME}{ModelHandler.MODEL_ALIAS}"
-        )
+        return f"api_{model_name.value}_{ENV_SHORT_NAME}{ModelHandler.MODEL_ALIAS}"