QTZPOC-13

QTZPOC-13: Centralize MLflow joblib fallback logic, ensuring consistent model version resolution by reusing the initial run ID, improving logging, and updating dependencies.

tests/laborious/utils/repository/test_model_repository.py

@@ -1903,6 +1903,28 @@ async def test_load_transform_model_fallback_recovers_load(
     joblib_mock.load.assert_called_once_with(str(tmp_path / 'model.pkl'))
 
 
+@patch(JOBLIB_PATH)
+@pytest.mark.asyncio
+async def test_load_transform_model_fallback_reuses_resolved_run(
+    joblib_mock, mlflow, mlflow_repository, tmp_path
+):
+    """A promotion between both steps must not change the run served."""
+    mlflow.sklearn.load_model.side_effect = Exception('flavor boom')
+    mlflow_repository.get_model_run_id = MagicMock(side_effect=['run-A', 'run-B'])
+    mlflow_repository.get_model_uri = MagicMock()
+    mlflow_repository.client.download_artifacts.return_value = build_artifact_dir(
+        tmp_path, 'MLmodel', 'model.pkl'
+    )
+
+    result = await mlflow_repository.load_transform_model('test_model', {}, 'sklearn')
+
+    assert result is joblib_mock.load.return_value
+    mlflow_repository.get_model_run_id.assert_called_once_with(
+        model_name='test_model', stage='Production'
+    )
+    mlflow_repository.client.download_artifacts.assert_called_once_with('run-A', 'data_model')
+
+
 @patch(JOBLIB_PATH)
 @pytest.mark.asyncio
 async def test_download_model_wrapper_fallback_reuses_downloaded_dir(