From 2c6740a60029f0bab6d339285c5a2a970e9c814c Mon Sep 17 00:00:00 2001 From: jdebacker Date: Sat, 2 Dec 2023 15:51:39 -0500 Subject: [PATCH] format --- ogzaf/demographics.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/ogzaf/demographics.py b/ogzaf/demographics.py index 55a3fd2..ad41334 100755 --- 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])