format

jdebacker · Dec 2, 2023 · 2c6740a · 2c6740a
1 parent 4bb5123
commit 2c6740a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/ogzaf/demographics.py b/ogzaf/demographics.py
@@ -790,7 +790,9 @@ def pop_rebin(curr_pop_dist, totpers_new):
     return curr_pop_new
 
 
-def get_imm_resid(totpers, start_year=2021, end_year=None, download=False, graph=False):
+def get_imm_resid(
+    totpers, start_year=2021, end_year=None, download=False, graph=False
+):
     """
     Calculate immigration rates by age as a residual given population levels in
     different periods, then output average calculated immigration rate. We have
@@ -846,7 +848,9 @@ def get_imm_resid(totpers, start_year=2021, end_year=None, download=False, graph
     pop_2020_EpS = pop_rebin(pop_2020, totpers)
     pop_2021_EpS = pop_rebin(pop_2021, totpers)
 
-    fert_rates = get_fert(totpers, start_year=start_year, end_year=end_year, download=download)
+    fert_rates = get_fert(
+        totpers, start_year=start_year, end_year=end_year, download=download
+    )
     mort_rates, infmort_rate = get_mort(
         totpers, start_year=start_year, end_year=end_year, download=download
     )
@@ -957,7 +961,9 @@ def get_pop_objs(E, S, T, curr_year, download=False, GraphDiag=False):
         E + S, start_year=hardcode_start_year, download=download
     )
     mort_rates_S = mort_rates[-S:]
-    imm_rates_orig = get_imm_resid(E + S, start_year=hardcode_start_year, download=download)
+    imm_rates_orig = get_imm_resid(
+        E + S, start_year=hardcode_start_year, download=download
+    )
     OMEGA_orig = np.zeros((E + S, E + S))
     OMEGA_orig[0, :] = (1 - infmort_rate) * fert_rates
     OMEGA_orig[1:, :-1] += np.diag(1 - mort_rates[:-1])