Fix travis build

 - Install keops
 - catch modulenotfound error, since this will happen on cpu-only machines
 - pep8

.travis.yml

@@ -17,6 +17,7 @@ install:
   - conda install -y flake8 codecov
   # check manifest needs conda-forge:
   - conda install -y -c conda-forge check-manifest
+  - pip install -e ./keops/
   - pip install -e .
 script:
   - pytest --cov-report=term-missing --cov=falkon --cov-config setup.cfg

falkon/ooc_ops/__init__.py

@@ -1,7 +1,7 @@
 try:
     from .ooc_lauum import gpu_lauum
     from .ooc_potrf import gpu_cholesky
-except OSError:
+except (OSError, ModuleNotFoundError):
     # No GPU
     gpu_lauum = None
     gpu_cholesky = None

falkon/tests/test_chol_prec.py

@@ -154,7 +154,7 @@ def test_cuda_start(self, mat, kernel, gram, dtype, order):
 
         np.testing.assert_allclose(prec.dT.numpy(), gpu_prec.dT.cpu().numpy(), rtol=rtol)
         np.testing.assert_allclose(prec.dA.numpy(), gpu_prec.dA.cpu().numpy(), rtol=rtol)
-        np.testing.assert_allclose(prec.fC.numpy(), gpu_prec.fC.cpu().numpy(), rtol=rtol*10)
+        np.testing.assert_allclose(prec.fC.numpy(), gpu_prec.fC.cpu().numpy(), rtol=rtol * 10)
         assert gpu_prec.fC.device == gpu_mat.device, "Device changed unexpectedly"
 
         assert_invariant_on_TT(gpu_prec, gpu_gram, tol=rtol)

falkon/tests/test_fmmv.py

@@ -26,7 +26,6 @@
 ]
 
 
-
 def choose_on_dtype(dtype):
     if dtype == np.float64 or dtype == torch.float64:
         return 1e-12
@@ -261,6 +260,7 @@ def test_incorrect_dev_setting(self, A, B, v, w, kernel, e_dfmmv1, expected_fmmv
 class TestKeops:
     basic_options = FalkonOptions(debug=True, compute_arch_speed=False, keops_active="force",
                                   max_cpu_mem=max_mem_dense, max_gpu_mem=max_mem_dense)
+
     @pytest.mark.parametrize("Ao,Adt,Bo,Bdt,vo,vdt", [
         ("C", np.float32, "C", np.float32, "C", np.float32),
         ("C", np.float64, "C", np.float64, "C", np.float64),
@@ -311,8 +311,8 @@ def test_gpu_inputs_fail(self, A, B, v, kernel, expected_fmmv):
 
 
 class TestSparse:
-    basic_options = FalkonOptions(debug=True, compute_arch_speed=False, 
-        max_cpu_mem=max_mem_sparse, max_gpu_mem=max_mem_sparse)
+    basic_options = FalkonOptions(debug=True, compute_arch_speed=False,
+                                  max_cpu_mem=max_mem_sparse, max_gpu_mem=max_mem_sparse)
     # sparse_dim and sparse_density result in sparse matrices with m and n non-zero entries.
     sparse_dim = 10_000
     sparse_density = 1e-4

falkon/tests/test_ooc_lauum.py

@@ -10,9 +10,9 @@
 from falkon.utils import decide_cuda
 from falkon.utils.helpers import sizeof_dtype
 from falkon.utils.tensor_helpers import move_tensor
+from falkon.ooc_ops.ooc_utils import calc_block_sizes3
 
 if decide_cuda():
-    from falkon.ooc_ops.ooc_utils import calc_block_sizes3
     from falkon.ooc_ops.ooc_lauum import gpu_lauum
     # noinspection PyUnresolvedReferences
     from falkon.ooc_ops.cuda import cuda_lauum_lower