SIENTIAPDE-1241: Suppress sklearn FutureWarning regarding 'squared' deprecation.
This commit is contained in:
@@ -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):
|
||||
|
||||
Reference in New Issue
Block a user