Set state for all parameters to fix bug

jdebacker · Jan 4, 2021 · f75d6b7 · f75d6b7
1 parent 4582583
commit f75d6b7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/taxcalc/parameters.py b/taxcalc/parameters.py
@@ -424,6 +424,7 @@ def adjust_with_indexing(self, params_or_path, **kwargs):
         # Re-instate ops.
         self.label_to_extend = label_to_extend
         self.array_first = array_first
+        self.set_state()
 
         # Filter out "-indexed" params.
         nonindexed_params = {
@@ -432,10 +433,6 @@ def adjust_with_indexing(self, params_or_path, **kwargs):
             if param not in index_affected
         }
 
-        needs_reset = set(needs_reset) - set(nonindexed_params.keys())
-        if needs_reset:
-            self._set_state(params=needs_reset)
-
         # 3. Do adjustment for all non-indexing related parameters.
         adj = super().adjust(nonindexed_params, **kwargs)