From c0a799356e975cf315147cf92f5c34c0537a8bf0 Mon Sep 17 00:00:00 2001 From: LaurentM Pass Date: Tue, 17 Sep 2024 15:21:06 +0200 Subject: [PATCH] chore: move @ for model alias outside alias_name --- .../compliance/api/src/pcpapillon/utils/model_handler.py | 4 ++-- 1 file changed, 2 insertions(+), 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 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}"