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():
|
class MLFlowRepository():
|
||||||
def __init__(self, host: str, username: str, password: str, logger: Logger):
|
def __init__(self, host: str, username: str, password: str, logger: Logger):
|
||||||
|
|
||||||
|
|
||||||
<< << << < HEAD
|
|
||||||
# set tracking uri
|
# set tracking uri
|
||||||
mlflow.set_tracking_uri(host)
|
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_USERNAME"] = username
|
||||||
environ["MLFLOW_TRACKING_PASSWORD"] = password
|
environ["MLFLOW_TRACKING_PASSWORD"] = password
|
||||||
|
|||||||
Reference in New Issue
Block a user