From 59d7e190d0d098293b79cda170233d563ae19fc4 Mon Sep 17 00:00:00 2001
From: LaurentM Pass <laurent.montier@passculture.app>
Date: Tue, 17 Sep 2024 14:22:48 +0200
Subject: [PATCH] fix: change model names

---
 apps/fraud/compliance/api/src/pcpapillon/utils/constants.py   | 4 ++--
 .../compliance/api/src/pcpapillon/utils/model_handler.py      | 4 +---
 2 files changed, 3 insertions(+), 5 deletions(-)

diff --git a/apps/fraud/compliance/api/src/pcpapillon/utils/constants.py b/apps/fraud/compliance/api/src/pcpapillon/utils/constants.py
index e248b65c..ea818539 100644
--- 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
index 4d84d9b5..8a07a06f 100644
--- 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}"