diff --git a/scripts/run_training_test.py b/scripts/run_training_test.py index cb4ccc2..a3f06a0 100644 --- a/scripts/run_training_test.py +++ b/scripts/run_training_test.py @@ -82,8 +82,8 @@ def upload_to_minio(source_path: Path) -> str: object_name = f'test-model-data-{timestamp}.csv' target_uri = f'{MINIO_ALIAS}/{MINIO_BUCKET}/{object_name}' - subprocess.run( - ['mc', 'cp', str(source_path), target_uri], + subprocess.run( # noqa: S603 + ['mc', 'cp', str(source_path), target_uri], # noqa: S607 check=True, ) return object_name diff --git a/tests/utils/repository/test_training_repository.py b/tests/utils/repository/test_training_repository.py index eb62618..7707bda 100644 --- a/tests/utils/repository/test_training_repository.py +++ b/tests/utils/repository/test_training_repository.py @@ -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