Merge pull request #861 from PyPSA/fix_clustering

fix clustering bug due to missing countries in n.buses
2024-01-18 19:34:59 +01:00 · 2024-01-18 19:34:59 +01:00 · 85ecb1355a
commit 85ecb1355a
parent dd1e07ef60 c0c6ce56e6
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.fillna("na") == "na"
    if c_nan_b.sum() > 0:
        c_tag = _get_country(buses.loc[c_nan_b])
        c_tag.loc[~c_tag.isin(countries)] = np.nan