SIENTIAPDE-1241: Linted files
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user