fmt

Author: guillaumehu

tests/geometry/geodesic_test.py

@@ -25,6 +25,7 @@
 from ott.problems.linear import linear_problem
 from ott.solvers.linear import sinkhorn
 
+
 def random_graph(
     n: int,
     p: float = 0.3,
@@ -271,12 +272,16 @@ def test_sparse_geodesic(self, normalize):
     n = 20
     G = random_graph(n, p=0.5)
     G_sparse = sparse.BCOO.fromdense(G)
-    geom = geodesic.Geodesic.from_graph(G_sparse, t=5.0, order=10, normalize=normalize)
+    geom = geodesic.Geodesic.from_graph(
+        G_sparse, t=5.0, order=10, normalize=normalize
+    )
     kernel_matrix = geom.kernel_matrix
 
     gh_heat_kernel = exact_heat_kernel(G, normalize=normalize, t=5.0)
 
     assert isinstance(kernel_matrix, sparse.BCOO)
 
     kernel_matrix = kernel_matrix.todense()
-    np.testing.assert_allclose(kernel_matrix, gh_heat_kernel, rtol=1e-1, atol=1e-1)
+    np.testing.assert_allclose(
+        kernel_matrix, gh_heat_kernel, rtol=1e-1, atol=1e-1
+    )