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

SIENTIAPDE-1243: Fix: Corrected dataframe inference and mlflow repository tests, and updated activity method calls in format and export prediction tests.

Browse Source
This commit is contained in:
Bruno Domingues
2025-10-01 17:54:02 -03:00
parent ef3c2a0c8a
commit 07e699aae2
3 changed files with 7 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/laborious/utils/repository/test_model_repository.py
View File

@@ -194,7 +194,7 @@ def test_get_experiment_success(mlflow, mlflow_repository):
output = mlflow_repository.get_experiment('test')
assert output == 0
assert output == '0'
@patch('model_manager.utils.repository.model_repository.mlflow')
@@ -245,7 +245,7 @@ def test_get_experiment_last_run_error(mlflow, mlflow_repository):
@patch('model_manager.utils.repository.model_repository.mlflow.sklearn')
@patch('model_manager.utils.repository.model_repository.mlflow.set_experiment')
def test_create_model_experiment(set_experiment, sklearn, mlflow_repository):
mlflow_repository.model_serving.get_model_run_id = MagicMock(return_value='0')
mlflow_repository.model_serving.get_model_info = MagicMock(return_value='0')
mlflow_repository.model_serving.get_model_uri = MagicMock(return_value='test')
mlflow_repository.get_experiment_by_run_id = MagicMock()
@@ -268,9 +268,7 @@ def test_create_model_experiment(set_experiment, sklearn, mlflow_repository):
output = mlflow_repository.create_model_experiment('test', data)
mlflow_repository.model_serving.get_model_run_id.assert_called_once_with(
'test', stage='Production'
)
mlflow_repository.model_serving.get_model_info.assert_called_once_with('test')
mlflow_repository.model_serving.get_model_uri.assert_called_once_with('0', prediction=False)
sklearn.load_model.assert_has_calls(