Fix indices for prediction intervals (#4154)

eccabay · web-flow · commit 83cfac867188 · 2023-04-18T15:38:20.000Z
* Add extra index fix
diff --git a/docs/source/release_notes.rst b/docs/source/release_notes.rst
@@ -6,6 +6,7 @@ Release Notes
         * Remove extra naive pipelines :pr:`4142`
     * Fixes
         * Fixed usage of codecov after uploader deprecation :pr:`4144`
+        * Fixed issue where prediction intervals were becoming NaNs due to index errors :pr:`4154`
     * Changes
     * Documentation Changes
     * Testing Changes
diff --git a/evalml/pipelines/time_series_regression_pipeline.py b/evalml/pipelines/time_series_regression_pipeline.py
@@ -252,6 +252,8 @@ def get_prediction_intervals(
                 y_train=y_train,
                 calculating_residuals=True,
             )
+            if self.component_graph.has_dfs:
+                predictions_train.index = y_train.index
             residuals = y_train - predictions_train
             std_residual = np.sqrt(np.sum(residuals**2) / len(residuals))
 

Original file line number	Diff line number	Diff line change
`@@ -252,6 +252,8 @@ def get_prediction_intervals(`
`252`	`252`	`y_train=y_train,`
`253`	`253`	`calculating_residuals=True,`
`254`	`254`	`)`
	`255`	`+ if self.component_graph.has_dfs:`
	`256`	`+ predictions_train.index = y_train.index`
`255`	`257`	`residuals = y_train - predictions_train`
`256`	`258`	`std_residual = np.sqrt(np.sum(residuals**2) / len(residuals))`
`257`	`259`