Blackified.

EconForge · Mar 30, 2021 · bde421e · bde421e
1 parent 37214cb
commit bde421e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/dolo/algos/simulations.py b/dolo/algos/simulations.py
@@ -24,11 +24,10 @@ def response(model, dr, varname, T=40, impulse: float = None):
     e1 = numpy.zeros(len(model.symbols["exogenous"]))
     e1[i_exo] = impulse
 
-
     exogenous = model.exogenous
     print(exogenous)
     print(T, e1)
-    m_simul = model.exogenous.response(T-1, e1) # this is an xarray T x V
+    m_simul = model.exogenous.response(T - 1, e1)  # this is an xarray T x V
     m_simul = m_simul.expand_dims("N")
     m_simul = m_simul.transpose("T", "N", "V").data
 

diff --git a/dolo/tests/test_irf.py b/dolo/tests/test_irf.py
@@ -6,7 +6,8 @@ def test_irf():
     model = yaml_import("examples/models/rbc.yaml")
     dr = perturb(model).dr
 
-    irf = response(model, dr, 'e_z')
+    irf = response(model, dr, "e_z")
     print(irf)
 
-test_irf()
+
+test_irf()