[BUG] fix absolute errorbar (#1579)

MartinoMensio · web-flow · commit 7d64fcebbb7b · 2025-03-21T01:26:02.000+05:30
This PR fixes an issue that occurrs when `max_prediction_length=1`: new matplotlib versions do not allow negative yerr in the errorbar, resulting in `ValueError: 'yerr' must not contain negative values`. fixes #1145
diff --git a/pytorch_forecasting/models/base/_base_model.py b/pytorch_forecasting/models/base/_base_model.py
@@ -1255,7 +1255,7 @@ def plot_prediction(
                     ax.errorbar(
                         x_pred,
                         y[[-n_pred]],
-                        yerr=quantiles - y[-n_pred],
+                        yerr=np.absolute(quantiles - y[-n_pred]),
                         c=pred_color,
                         capsize=1.0,
                     )

Original file line number	Diff line number	Diff line change
`@@ -1255,7 +1255,7 @@ def plot_prediction(`
`1255`	`1255`	`ax.errorbar(`
`1256`	`1256`	`x_pred,`
`1257`	`1257`	`y[[-n_pred]],`
`1258`		`- yerr=quantiles - y[-n_pred],`
	`1258`	`+ yerr=np.absolute(quantiles - y[-n_pred]),`
`1259`	`1259`	`c=pred_color,`
`1260`	`1260`	`capsize=1.0,`
`1261`	`1261`	`)`