Fix bug in IRF

jessegrabowski · Nov 19, 2024 · c8f79db · c8f79db
1 parent 5b94990
commit c8f79db
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/gEconpy/model/model.py b/gEconpy/model/model.py
@@ -1701,7 +1701,9 @@ def _make_shock_dict(shocks, shock_size=None, Q=None):
 
     elif return_individual_shocks and shock_trajectory is not None:
         for i, shock_name in enumerate(shock_names):
-            data[i] = _simulate(shock_trajectory[i])
+            traj = np.zeros_like(shock_trajectory)
+            traj[i] = shock_trajectory[i]
+            data[i] = _simulate(traj)
 
     else:
         traj = _create_shock_trajectory(