Turn off calc_stable_points warning when irrelevant

PerfForesightConsumerType.calc_stable_points was raising a warning in child models when cycles=0 and T_cycle=1, even though the method is inapplicable to them. Now they won't.
econ-ark · Sep 3, 2024 · 8dfa346 · 8dfa346
1 parent 1a34e73
commit 8dfa346
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/HARK/ConsumptionSaving/ConsIndShockModel.py b/HARK/ConsumptionSaving/ConsIndShockModel.py
@@ -1854,6 +1854,12 @@ def calc_stable_points(self):
         -------
         None
         """
+        # Child classes should not run this method
+        is_perf_foresight = type(self) is PerfForesightConsumerType
+        is_ind_shock = type(self) is IndShockConsumerType
+        if not (is_perf_foresight or is_ind_shock):
+            return
+
         infinite_horizon = self.cycles == 0
         single_period = self.T_cycle = 1
         if not infinite_horizon: