gitea_admin/sientia-dataops-model-manager
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SIENTIAPDE-1241: Linted files

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

4
scripts/run_training_test.py
View File

@@ -82,8 +82,8 @@ def upload_to_minio(source_path: Path) -> str:
object_name = f'test-model-data-{timestamp}.csv' object_name = f'test-model-data-{timestamp}.csv'
target_uri = f'{MINIO_ALIAS}/{MINIO_BUCKET}/{object_name}' target_uri = f'{MINIO_ALIAS}/{MINIO_BUCKET}/{object_name}'
subprocess.run( subprocess.run( # noqa: S603
['mc', 'cp', str(source_path), target_uri], ['mc', 'cp', str(source_path), target_uri], # noqa: S607
check=True, check=True,
) )
return object_name return object_name

8
tests/utils/repository/test_training_repository.py
View File

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