SIENTIAPDE-1231

Refactor tests and update model_repository.py for clarity and consistency

- Added a blank line in model_repository.py for improved readability.
- Adjusted formatting in test_mlflow.py to streamline assertions.
- Updated ensure_bucket_exists method tests in test_minio_repository.py to reflect the new return value of None instead of True.
This commit is contained in:
vitor-aignosi
2025-10-16 10:15:53 -03:00
parent d59833a5e5
commit de47820c4a
3 changed files with 5 additions and 6 deletions

View File

@@ -35,6 +35,7 @@ PREDICTION_COMPRESSED_PATH = 'artifacts/stacking_model.pkl'
INVALID_FLAVOR_MESSAGE = "Invalid flavor. Use 'sklearn' or 'pyfunc' or 'pytorch'."
def force_memory_release(logger: Logger):
gc.collect()
@@ -629,7 +630,7 @@ class MLFlowRepository:
# Aligns data with treated data indexes to get target variable
aligned_data = data.loc[treated_data.index]
aligned_series = aligned_data[target_name]
retrain_dataset = pd.merge( # NOSONAR
retrain_dataset = pd.merge( # NOSONAR
treated_data, aligned_series, left_index=True, right_index=True
)
else: