Skip to content

Commit 512baf1

Browse files
committed
format
1 parent a53a1dc commit 512baf1

File tree

1 file changed

+13
-4
lines changed

1 file changed

+13
-4
lines changed

ogcore/demographics.py

Lines changed: 13 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -147,7 +147,9 @@ def get_fert(
147147
"68", country_id=country_id, start_year=start_year, end_year=end_year
148148
)
149149
if download_path:
150-
df.to_csv(os.path.join(download_path, "raw_fert_data_UN.csv"), index=False)
150+
df.to_csv(
151+
os.path.join(download_path, "raw_fert_data_UN.csv"), index=False
152+
)
151153
# CLean and rebin data
152154
for y in range(start_year, end_year + 1):
153155
df_y = df[(df.age >= min_age) & (df.age <= max_age) & (df.year == y)]
@@ -235,7 +237,9 @@ def get_mort(
235237
"80", country_id=country_id, start_year=start_year, end_year=end_year
236238
)
237239
if download_path:
238-
df.to_csv(os.path.join(download_path, "raw_mort_data_UN.csv"), index=False)
240+
df.to_csv(
241+
os.path.join(download_path, "raw_mort_data_UN.csv"), index=False
242+
)
239243
# CLean and rebin data
240244
for y in range(start_year, end_year + 1):
241245
df_y = df[(df.age >= min_age) & (df.age <= max_age) & (df.year == y)]
@@ -350,7 +354,10 @@ def get_pop(
350354
end_year=start_year - 1,
351355
)
352356
if download_path:
353-
pre_pop_data.to_csv(os.path.join(download_path, "raw_pre_pop_data_UN.csv"), index=False)
357+
pre_pop_data.to_csv(
358+
os.path.join(download_path, "raw_pre_pop_data_UN.csv"),
359+
index=False,
360+
)
354361
pre_pop_sample = pre_pop_data[
355362
(pre_pop_data["age"] >= min_age)
356363
& (pre_pop_data["age"] <= max_age)
@@ -404,7 +411,9 @@ def get_pop(
404411
+ 2, # note go to + 2 because needed to infer immigration for end_year
405412
)
406413
if download_path:
407-
pop_data.to_csv(os.path.join(download_path, "raw_pop_data_UN.csv"), index=False)
414+
pop_data.to_csv(
415+
os.path.join(download_path, "raw_pop_data_UN.csv"), index=False
416+
)
408417
# CLean and rebin data
409418
for y in range(start_year, end_year + 2):
410419
pop_data_sample = pop_data[

0 commit comments

Comments
 (0)