shock_size should never be scalar

pymc-devs · Jun 26, 2024 · 83ebb80 · 83ebb80
1 parent 94790c7
commit 83ebb80
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pymc_experimental/statespace/core/statespace.py b/pymc_experimental/statespace/core/statespace.py
@@ -1664,7 +1664,9 @@ def impulse_response_function(
                     init_shock = pm.MvNormal("initial_shock", mu=0, cov=Q, dims=[SHOCK_DIM])
                 else:
                     init_shock = pm.Deterministic(
-                        "initial_shock", pt.as_tensor_variable(shock_size), dims=[SHOCK_DIM]
+                        "initial_shock",
+                        pt.as_tensor_variable(np.atleast_1d(shock_size)),
+                        dims=[SHOCK_DIM],
                     )
                 shock_trajectory = pt.set_subtensor(shock_trajectory[0], init_shock)