base_network: fix pandas 2.1 issue

2023-12-18 09:57:05 +01:00 · 2023-12-18 09:57:05 +01:00 · 7e543ad0cc
commit 7e543ad0cc
parent 84d38228ae
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.isnull()
+    c_nan_b = buses.country == '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