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 8a07a06f..34611512 100644 --- 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}"