fix bus filter for buses without countries

2024-01-12 16:50:17 +01:00 · 2024-01-12 16:50:17 +01:00 · 8565c124c0
commit 8565c124c0
parent a6edce58fb
1 changed files with 1 additions and 1 deletions
--- a/scripts/base_network.py
+++ b/scripts/base_network.py
@ -560,7 +560,7 @@ def _set_countries_and_substations(n, config, country_shapes, offshore_shapes):
        ~buses["under_construction"]
    )

-    c_nan_b = buses.country == "na"
+    c_nan_b = buses.country.isna()
    if c_nan_b.sum() > 0:
        c_tag = _get_country(buses.loc[c_nan_b])
        c_tag.loc[~c_tag.isin(countries)] = np.nan