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:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user