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