SIENTIAPDE-1241: Linted files

This commit is contained in:
Kou-Kinoshita
2025-10-29 11:57:40 -03:00
parent 5e763a8a64
commit 1f6765dc86
2 changed files with 7 additions and 5 deletions

View File

@@ -1,6 +1,6 @@
"""Unit tests for TrainingRepository."""
from unittest.mock import MagicMock, Mock, patch
from unittest.mock import MagicMock
import numpy as np
import pandas as pd
@@ -236,9 +236,10 @@ class TestInitScalerDict:
def test_init_scaler_dict_with_minmax_scaler(self, training_repo, sample_params):
"""Test scaler dict initialization with MinMaxScaler."""
from model_manager.sientia.models import DataPreprocessor
from sientia_do.operations.normalization import MinMaxScaler
from model_manager.sientia.models import DataPreprocessor
sample_params.use_scaler = True
# Create mock MinMaxScaler
@@ -268,9 +269,10 @@ class TestInitScalerDict:
def test_init_scaler_dict_with_z_scaler(self, training_repo, sample_params):
"""Test scaler dict initialization with Z_Scaler."""
from model_manager.sientia.models import DataPreprocessor
from sientia_do.operations.normalization import Z_Scaler
from model_manager.sientia.models import DataPreprocessor
sample_params.use_scaler = True
# Create mock Z_Scaler