Skip to content

Commit 75ef730

Browse files
committed
flake8 formating
1 parent 78b57d6 commit 75ef730

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

stochman/discretized_manifold.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,7 +294,9 @@ def connecting_geodesic(self, p1, p2, curve=None):
294294
with torch.no_grad():
295295
idx1 = self._grid_point(p1[b].unsqueeze(0))
296296
idx2 = self._grid_point(p2[b].unsqueeze(0))
297-
path = nx.shortest_path(self.G, source=idx1, target=idx2, weight='weight') # list with N elements
297+
path = nx.shortest_path(
298+
self.G, source=idx1, target=idx2, weight='weight'
299+
) # list with N elements
298300
weights = [self.G.edges[path[k], path[k + 1]]['weight'] for k in range(len(path) - 1)]
299301
mesh = torch.meshgrid(*self.grid, indexing='ij')
300302
raw_coordinates = [m.flatten()[path[1:-1]].view(-1, 1) for m in mesh]

0 commit comments

Comments
 (0)