make sure updated dummy target has correct dimensions

Dekermanjian · Dekermanjian · commit 67bd44e2e23b · 2025-05-02T08:54:24.000-06:00
diff --git a/pymc_extras/statespace/core/statespace.py b/pymc_extras/statespace/core/statespace.py
@@ -2072,7 +2072,7 @@ def forecast(
             for name in self.data_names:
                 if name in scenario.keys():
                     pm.set_data(
-                        {"data": np.zeros_like(scenario[name])},
+                        {"data": np.zeros((len(forecast_index), self.k_endog))},
                         coords={"data_time": np.arange(len(forecast_index))},
                     )
                     break

Original file line number	Diff line number	Diff line change
`@@ -2072,7 +2072,7 @@ def forecast(`
`2072`	`2072`	`for name in self.data_names:`
`2073`	`2073`	`if name in scenario.keys():`
`2074`	`2074`	`pm.set_data(`
`2075`		`- {"data": np.zeros_like(scenario[name])},`
	`2075`	`+ {"data": np.zeros((len(forecast_index), self.k_endog))},`
`2076`	`2076`	`coords={"data_time": np.arange(len(forecast_index))},`
`2077`	`2077`	`)`
`2078`	`2078`	`break`