Skip to content

Commit faab531

Browse files
committed
refactor: increase robustness of extract_custom_transmission_projetcs
1 parent 0a209a3 commit faab531

1 file changed

Lines changed: 6 additions & 3 deletions

File tree

scripts/cba/clean_projects.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -266,15 +266,18 @@ def extract_custom_transmission_projects(
266266
Curated list of custom projects.
267267
"""
268268
custom_transmission_projects = (
269-
pd.read_csv(custom_transmission_path)
269+
pd.read_csv(
270+
custom_transmission_path,
271+
true_values=["TRUE", "True", "true", "1"],
272+
false_values=["FALSE", "False", "false", "0"],
273+
)
270274
.assign(border=lambda df: df.bus0 + "-" + df.bus1)
271-
.drop(["source", "further description"], axis=1)
275+
.drop(["source", "further description"], axis=1, errors="ignore")
272276
)
273277

274278
custom_transmission_projects = remove_unclear_border(
275279
custom_transmission_projects, existing_buses
276280
)
277-
custom_transmission_projects = remove_no_capacity(custom_transmission_projects)
278281

279282
return custom_transmission_projects
280283

0 commit comments

Comments
 (0)