@@ -129,7 +129,7 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
129129 - converters (GeoDataFrame): GeoDataFrame containing converter data with geometries.
130130 - transformers (GeoDataFrame): GeoDataFrame containing transformer data with geometries.
131131 """
132-
132+
133133 buses_cols = [
134134 "Bus" ,
135135 "v_nom" ,
@@ -139,7 +139,9 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
139139 "tags" ,
140140 "geometry" ,
141141 ]
142- buses = network .buses .reset_index ()[[c for c in buses_cols if c in network .buses .columns ]]
142+ buses = network .buses .reset_index ()[
143+ [c for c in buses_cols if c in network .buses .columns ]
144+ ]
143145 buses ["geometry" ] = buses .geometry .apply (lambda x : loads (x ))
144146 buses = gpd .GeoDataFrame (buses , geometry = "geometry" , crs = crs )
145147
@@ -161,7 +163,9 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
161163 "tags" ,
162164 "geometry" ,
163165 ]
164- lines = network .lines .reset_index ()[[c for c in lines_cols if c in network .lines .columns ]]
166+ lines = network .lines .reset_index ()[
167+ [c for c in lines_cols if c in network .lines .columns ]
168+ ]
165169 # Create shapely linestring from geometry column
166170 lines ["geometry" ] = lines .geometry .apply (lambda x : loads (x ))
167171 lines = gpd .GeoDataFrame (lines , geometry = "geometry" , crs = crs )
@@ -181,7 +185,9 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
181185 links = (
182186 network .links [~ is_converter ]
183187 .reset_index ()
184- .rename (columns = {"voltage" : "v_nom" })[[c for c in links_cols if c in network .links .columns ]]
188+ .rename (columns = {"voltage" : "v_nom" })[
189+ [c for c in links_cols if c in network .links .columns ]
190+ ]
185191 )
186192 links ["geometry" ] = links .geometry .apply (lambda x : loads (x ))
187193 links = gpd .GeoDataFrame (links , geometry = "geometry" , crs = crs )
@@ -197,7 +203,9 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
197203 converters = (
198204 network .links [is_converter ]
199205 .reset_index ()
200- .rename (columns = {"voltage" : "v_nom" })[[c for c in converters_cols if c in network .links .columns ]]
206+ .rename (columns = {"voltage" : "v_nom" })[
207+ [c for c in converters_cols if c in network .links .columns ]
208+ ]
201209 )
202210 converters ["geometry" ] = converters .geometry .apply (lambda x : loads (x ))
203211 converters = gpd .GeoDataFrame (converters , geometry = "geometry" , crs = crs )
@@ -211,7 +219,9 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
211219 "s_nom" ,
212220 "geometry" ,
213221 ]
214- transformers = network .transformers .reset_index ()[[c for c in transformers_cols if c in network .transformers .columns ]]
222+ transformers = network .transformers .reset_index ()[
223+ [c for c in transformers_cols if c in network .transformers .columns ]
224+ ]
215225 transformers ["geometry" ] = transformers .geometry .apply (lambda x : loads (x ))
216226 transformers = gpd .GeoDataFrame (transformers , geometry = "geometry" , crs = crs )
217227
@@ -421,7 +431,7 @@ def create_geometries(network, is_converter, crs=GEO_CRS):
421431 name = "Links (DC)" ,
422432 zindex = 115 ,
423433 )
424-
434+
425435 if not transformers .empty :
426436 map = transformers .explore (
427437 color = "orange" ,
0 commit comments