Little sim fix.

OpenSourceEconomics · Nov 27, 2024 · 47993f4 · 47993f4
1 parent 2f73ac9
commit 47993f4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/dcegm/simulation/simulate.py b/src/dcegm/simulation/simulate.py
@@ -129,8 +129,9 @@ def simulate_all_periods(
         key: np.vstack([sim_dict[key], final_period_dict[key]])
         for key in sim_dict.keys()
     }
-    if "dummy_exog" not in model_sim["model_structure"]["exog_states_names"]:
-        result.pop("dummy_exog")
+    if "dummy_exog" in model_structure_solution["exog_states_names"]:
+        if "dummy_exog" not in model_sim["model_structure"]["exog_states_names"]:
+            result.pop("dummy_exog")
 
     return result