diff --git a/model_manager/utils/repository/model_repository.py b/model_manager/utils/repository/model_repository.py index 45068ee..d9617f0 100644 --- a/model_manager/utils/repository/model_repository.py +++ b/model_manager/utils/repository/model_repository.py @@ -11,6 +11,7 @@ and logging model runs to MLFlow. import os import shutil +import warnings from datetime import datetime from os import makedirs, path @@ -22,6 +23,9 @@ from model_manager.sientia.model_serving import ModelServing # type: ignore[imp from model_manager.sientia.reports import Reports # type: ignore[import-untyped] from model_manager.utils.models.train_model_result import TrainModelResult +# Suppress sklearn FutureWarning about 'squared' deprecation without changing business logic +warnings.filterwarnings('ignore', category=FutureWarning, message=".*'squared' is deprecated.*") + class ModelRepository: def __init__(self, url, username, password, logger: Logger):