Skip to content

Commit e8106ca

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 5e10951 commit e8106ca

File tree

1 file changed

+17
-7
lines changed

1 file changed

+17
-7
lines changed

scripts/prepare_osm_network_release.py

Lines changed: 17 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)