Updated test

alteryx · Jan 9, 2024 · 189be9d · 189be9d
1 parent d482500
commit 189be9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...ml/tests/pipeline_tests/regression_pipeline_tests/test_multiseries_regression_pipeline.py b/...ml/tests/pipeline_tests/regression_pipeline_tests/test_multiseries_regression_pipeline.py
@@ -347,7 +347,7 @@ def test_time_series_pipeline_get_prediction_intervals(
     )
     X_train, y_train = X[:65], y[:65]
     X_validation, y_validation = X[65:], y[65:]
-    mock_X, _ = unstack_multiseries(
+    mock_X, mock_y = unstack_multiseries(
         X_train,
         y_train,
         series_id="series_id",
@@ -356,7 +356,7 @@ def test_time_series_pipeline_get_prediction_intervals(
     )
     mock_transform_return_value = (
         mock_X,
-        pd.DataFrame(np.random.rand(13, 5)),
+        mock_y,
     )
     with patch(
         "evalml.pipelines.components.transformers.preprocessing.stl_decomposer.STLDecomposer.transform",