ruff fixes

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 ===============
 # =======================================================