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:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user