diff --git a/scripts/simplify_network.py b/scripts/simplify_network.py index 8663faa3..4830b87b 100644 --- a/scripts/simplify_network.py +++ b/scripts/simplify_network.py @@ -361,8 +361,8 @@ if __name__ == "__main__": if snakemake.wildcards.simpl: n, cluster_map = cluster(n, int(snakemake.wildcards.simpl)) busmaps.append(cluster_map) - - n.buses = n.buses.drop(['substation_lv', 'substation_off', 'under_construction'], axis=1) + else: + n.buses = n.buses.drop(['symbol', 'tags', 'under_construction', 'substation_lv', 'substation_off'], axis=1) n.export_to_netcdf(snakemake.output.network) busmap_s = reduce(lambda x, y: x.map(y), busmaps[1:], busmaps[0])