SIENTIAPDE-1222

Update model configuration keys in MLFlowRepository for consistency

- Changed 'model_retention' to 'retention_minutes' and 'is_compressed' to 'compressed' in model configuration handling, ensuring alignment with updated configuration standards.
This commit is contained in:
vitor-aignosi
2025-09-16 12:34:56 -03:00
parent 73b03e6749
commit 12b56cb71f

View File

@@ -27,16 +27,8 @@ ARTIFACTS_PATH = "./tmp/artifacts"
class MLFlowRepository():
def __init__(self, host: str, username: str, password: str, logger: Logger):
<< << << < HEAD
# set tracking uri
mlflow.set_tracking_uri(host)
== == == =
self.model_serving = ModelServing(tracking_uri=host,
username=username, password=password,
logger=logger)
self.logger = logger
>>>>>> > eefd081(SIENTIAPDE-1222)
environ["MLFLOW_TRACKING_USERNAME"] = username
environ["MLFLOW_TRACKING_PASSWORD"] = password