From 8565c124c034f90436c5b37c21424d456d7e5e00 Mon Sep 17 00:00:00 2001 From: Philipp Glaum Date: Fri, 12 Jan 2024 16:50:17 +0100 Subject: [PATCH] fix bus filter for buses without countries --- scripts/base_network.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/base_network.py b/scripts/base_network.py index eeb87bf5..340b6f36 100644 --- 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