fix test import and README

Author: saroudant

.github/workflows/codecov.yml

@@ -3,22 +3,20 @@
 import scipy
 import torch
 from sobolev_alignment import KRRApprox
+from sobolev_alignment.krr_approx import FALKON_IMPORTED
 
-n_samples = 2000
 n_samples_valid = 50
 n_genes = 100
 n_latent = 7
+n_samples = 2000
 penalization = 0.0001
 pearson_threshold = 0.99
 M = 500
 
 
 @pytest.fixture(scope="module")
 def falkon_import():
-    try:
-        return True
-    except ImportError:
-        return False
+    return FALKON_IMPORTED
 
 
 @pytest.fixture(scope="module")
@@ -160,6 +158,9 @@ def test_all_sklearn_kernels(self):
 
     def test_all_falkon_kernels(self, falkon_import):
         if falkon_import:
+            print(">>>>>>\n\n\n\n")
+            print(falkon_import)
+            print("\n\n\n<<<<<")
             for kernel in KRRApprox.falkon_kernel:
                 KRRApprox(kernel=kernel, method="falkon")
         return True

tests/test_krr_approx.py

@@ -3,6 +3,7 @@
 import pytest
 from anndata import AnnData
 from sobolev_alignment import SobolevAlignment
+from sobolev_alignment.krr_approx import FALKON_IMPORTED
 
 n_samples = 500
 n_samples_valid = 50
@@ -15,10 +16,7 @@
 
 @pytest.fixture(scope="module")
 def falkon_import():
-    try:
-        return True
-    except ImportError:
-        return False
+    return FALKON_IMPORTED
 
 
 @pytest.fixture(scope="module")