might needed for flake8

gabelstein · Apr 15, 2024 · a435235 · a435235
1 parent b988e96
commit a435235
Showing 1 changed file with 9 additions and 7 deletions.
diff --git a/tests/test_embedding.py b/tests/test_embedding.py
@@ -12,9 +12,9 @@
 )
 
 from pyriemann.utils.kernel import (kernel,
-                                    kernel_euclid,
-                                    kernel_logeuclid,
-                                    kernel_riemann)
+                                    kernel_euclid, # noqa
+                                    kernel_logeuclid, # noqa
+                                    kernel_riemann) # noqa
 
 rembd = [SpectralEmbedding, LocallyLinearEmbedding]
 
@@ -154,16 +154,18 @@ def test_locally_linear_none_kernel(metric, get_mats):
     n_matrices, n_channels, n_components = 6, 3, 2
     covmats = get_mats(n_matrices, n_channels, "spd")
     kernel_fun = globals()[f'kernel_{metric}']
-    kfun = lambda X, Y=None, Cref=None, metric=None: kernel_fun(X, Y, Cref=Cref)
+
+    def kfun(X, Y=None, Cref=None, metric=None):
+        return kernel_fun(X, Y, Cref=Cref)
 
     embd = LocallyLinearEmbedding(metric=metric,
                                   n_components=n_components,
                                   kernel=kfun)
     covembd = embd.fit_transform(covmats)
 
     embd2 = LocallyLinearEmbedding(metric=metric,
-                                  n_components=n_components,
-                                  kernel=None)
+                                   n_components=n_components,
+                                   kernel=None)
     covembd2 = embd2.fit_transform(covmats)
 
-    assert np.array_equal(covembd, covembd2)
+    assert np.array_equal(covembd, covembd2)