SIENTIAPDE-1273

Update sientia-mlops-library dependency to version 0.40.6 and refactor get_prediction_data method calls for improved readability in model_repository.py and test_model_repository.py.
This commit is contained in:
vitor-aignosi
2025-11-19 14:54:50 -03:00
parent 684bc9fb31
commit 3311409885
3 changed files with 13 additions and 4 deletions

View File

@@ -865,7 +865,10 @@ async def test_fit_models_not_df_target_name_none_and_not_in_model(
prediction_model.fit.assert_called_once_with(pd_merge.return_value)
mlflow_repository.get_prediction_data.assert_called_once_with(
prediction_model, pd_merge.return_value, data_model.fit.return_value.target_variable, 'pyfunc'
prediction_model,
pd_merge.return_value,
data_model.fit.return_value.target_variable,
'pyfunc',
)
assert output == {