ruff fixes

Author: sbenthall

HARK/model.py

@@ -15,6 +15,7 @@
 from HARK.parser import math_text_to_lambda
 from typing import Any, Callable, Mapping, List, Union
 
+
 class Aggregate:
     """
     Used to designate a shock as an aggregate shock.
@@ -267,7 +268,11 @@ def get_dynamics(self):
         return self.dynamics
 
     def get_vars(self):
-        return list(self.shocks.keys()) + list(self.dynamics.keys()) + list(self.reward.keys())
+        return (
+            list(self.shocks.keys())
+            + list(self.dynamics.keys())
+            + list(self.reward.keys())
+        )
 
     def transition(self, pre, dr):
         """

HARK/simulation/monte_carlo.py

@@ -456,7 +456,7 @@ def __init__(
         self.initial = initial
 
         self.seed = seed  # NOQA
-        self.agent_count = agent_count # TODO: pass this in at block level
+        self.agent_count = agent_count  # TODO: pass this in at block level
         self.T_sim = T_sim
 
         # changes here from HARK.core.AgentType
@@ -534,20 +534,20 @@ def sim_one_period(self):
 
         shocks_now = draw_shocks(
             self.shocks,
-            np.zeros(self.agent_count) # TODO: stupid hack to remove age calculations.
-                                       # this needs a little more thought
-            )
+            np.zeros(self.agent_count),  # TODO: stupid hack to remove age calculations.
+            # this needs a little more thought
+        )
 
-        pre = self.calibration # for AgentTypeMC, this is conditional on age
-                               # TODO: generalize indexing into calibration. 
+        pre = self.calibration  # for AgentTypeMC, this is conditional on age
+        # TODO: generalize indexing into calibration.
 
         pre.update(self.vars_prev)
         pre.update(shocks_now)
 
         # Won't work for 3.8: self.parameters | self.vars_prev | shocks_now
-        
-        dr = self.dr # AgentTypeMC chooses rule by age;
-                     # that generalizes to age as a DR argument?
+
+        dr = self.dr  # AgentTypeMC chooses rule by age;
+        # that generalizes to age as a DR argument?
 
         post = simulate_dynamics(self.dynamics, pre, dr)
 
@@ -570,7 +570,7 @@ def sim_birth(self, which_agents):
         -------
         None
         """
-        
+
         initial_vals = draw_shocks(self.initial, np.zeros(which_agents.sum()))
 
         if np.sum(which_agents) > 0:
@@ -636,4 +636,4 @@ def clear_history(self):
         """
         for var_name in self.vars:
             self.history[var_name] = np.empty((self.T_sim, self.agent_count))
-            self.history[var_name].fill(np.nan)
+            self.history[var_name].fill(np.nan)

HARK/simulation/test_monte_carlo.py

@@ -204,4 +204,4 @@ def test_simulate(self):
             - history["c"][5]
         )
 
-        self.assertTrue((a1 == b1).all())
+        self.assertTrue((a1 == b1).all())

HARK/utilities.py

@@ -129,6 +129,7 @@ def apply_fun_to_vals(fun, vals):
     """
     return fun(*[vals[var] for var in signature(fun).parameters])
 
+
 # =======================================================
 # ================ Other useful functions ===============
 # =======================================================