Skip to content

Commit 80a05a2

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

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

graph_weather/models/gencast/graph/graph_builder.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -184,7 +184,7 @@ def _init_grid2mesh_graph(self):
184184

185185
# Create some edges according to distance between mesh and grid nodes.
186186
assert self._grid_lat is not None and self._grid_lon is not None
187-
(grid_indices, mesh_indices) = grid_mesh_connectivity.radius_query_indices(
187+
grid_indices, mesh_indices = grid_mesh_connectivity.radius_query_indices(
188188
grid_latitude=self._grid_lat,
189189
grid_longitude=self._grid_lon,
190190
mesh=self._mesh,
@@ -198,7 +198,7 @@ def _init_grid2mesh_graph(self):
198198
# Precompute structural node and edge features according to config options.
199199
# Structural features are those that depend on the fixed values of the
200200
# latitude and longitudes of the nodes.
201-
(senders_node_features, receivers_node_features, edge_features) = (
201+
senders_node_features, receivers_node_features, edge_features = (
202202
model_utils.get_bipartite_graph_spatial_features(
203203
senders_node_lat=self._grid_nodes_lat,
204204
senders_node_lon=self._grid_nodes_lon,
@@ -265,7 +265,7 @@ def _init_mesh2grid_graph(self):
265265

266266
# Create some edges according to how the grid nodes are contained by
267267
# mesh triangles.
268-
(grid_indices, mesh_indices) = grid_mesh_connectivity.in_mesh_triangle_indices(
268+
grid_indices, mesh_indices = grid_mesh_connectivity.in_mesh_triangle_indices(
269269
grid_latitude=self._grid_lat, grid_longitude=self._grid_lon, mesh=self._mesh
270270
)
271271

@@ -275,7 +275,7 @@ def _init_mesh2grid_graph(self):
275275

276276
# Precompute structural node and edge features according to config options.
277277
assert self._mesh_nodes_lat is not None and self._mesh_nodes_lon is not None
278-
(senders_node_features, receivers_node_features, edge_features) = (
278+
senders_node_features, receivers_node_features, edge_features = (
279279
model_utils.get_bipartite_graph_spatial_features(
280280
senders_node_lat=self._mesh_nodes_lat,
281281
senders_node_lon=self._mesh_nodes_lon,

0 commit comments

Comments
 (0)