From 42a7e82c82f313f0e8aa3a8d52b8390d805711c9 Mon Sep 17 00:00:00 2001 From: Bruno Domingues Date: Wed, 22 Oct 2025 18:28:13 -0300 Subject: [PATCH] SIENTIAPDE-1241: Suppress sklearn FutureWarning regarding 'squared' deprecation. --- model_manager/utils/repository/model_repository.py | 4 ++++ 1 file changed, 4 insertions(+) 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):