SIENTIAPDE-1241: Suppress sklearn FutureWarning regarding 'squared' deprecation.

This commit is contained in:
Bruno Domingues
2025-10-22 18:28:13 -03:00
parent 5789a13023
commit 42a7e82c82

View File

@@ -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):