QTZPOC-20

QTZPOC-20: Remove PyTorch from supported model flavors as it was never functional due to missing runtime dependencies.

e2e/conftest.py

@@ -546,10 +546,6 @@ def patch_mlflow(mock_mlflow_models, joblib_flavor_artifacts):
     mock_mlflow.pyfunc = MagicMock()
     mock_mlflow.pyfunc.load_model = MagicMock(side_effect=mock_pyfunc_load_model)
 
-    # Mock pytorch.load_model
-    mock_mlflow.pytorch = MagicMock()
-    mock_mlflow.pytorch.load_model = MagicMock(return_value=mock_mlflow_models['predict_model'])
-
     # Mock other mlflow methods that might be called
     mock_mlflow.set_tracking_uri = MagicMock()
     mock_mlflow.get_run = MagicMock(

e2e/scenarios.md

@@ -1124,7 +1124,7 @@ restores `sys.path` in teardown, because the flavor deliberately never removes w
 **Assertions**:
 - Exactly one prediction row, `prediction_status='Good'`, value matching the artifact's model
 - `download_artifacts` called once, with `'prediction_model'`
-- No `mlflow.{sklearn,pyfunc,pytorch}.load_model` call names this model
+- No `mlflow.{sklearn,pyfunc}.load_model` call names this model
 
 ---
 

e2e/test_predictions_batch_joblib_flavor.py

@@ -26,7 +26,7 @@ from laborious.workflows.predictions_batch import PredictionsBatch
 
 base_query = 'SELECT timestamp, variable, value, created_at FROM predictions_schema.laborious_data WHERE model_id = {model_id}'
 
-FLAVOR_LOADERS = ('sklearn', 'pyfunc', 'pytorch')
+FLAVOR_LOADERS = ('sklearn', 'pyfunc')
 
 
 def build_input_data(