From 83ebb80e24f8e988497c968c9b266749a6f16e7d Mon Sep 17 00:00:00 2001 From: Jesse Grabowski Date: Wed, 26 Jun 2024 23:00:28 +0800 Subject: [PATCH] `shock_size` should never be scalar --- pymc_experimental/statespace/core/statespace.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/pymc_experimental/statespace/core/statespace.py b/pymc_experimental/statespace/core/statespace.py index d35620eee..e7209ec36 100644 --- 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)