diff --git a/GPy/testing/kernel_tests.py b/GPy/testing/kernel_tests.py index e5bc5683..5f418674 100644 --- a/GPy/testing/kernel_tests.py +++ b/GPy/testing/kernel_tests.py @@ -14,10 +14,10 @@ from ..util.config import config verbose = 0 try: - from ..util import choleskies_cython - cython_choleskies_working = True + from ..kern.src import coregionalize_cython + cython_coregionalize_working = True except ImportError: - cython_choleskies_working = False + cython_coregionalize_working = False class Kern_check_model(GPy.core.Model): @@ -618,7 +618,7 @@ class KernelTestsNonContinuous(unittest.TestCase): kern = GPy.kern.Coregionalize(1, output_dim=3, active_dims=[-1]) self.assertTrue(check_kernel_gradient_functions(kern, X=self.X, X2=self.X2, verbose=verbose, fixed_X_dims=-1)) -@unittest.skipIf(not cython_choleskies_working,"Cython choleskies module has not been built on this machine") +@unittest.skipIf(not cython_coregionalize_working,"Cython coregionalize module has not been built on this machine") class Coregionalize_cython_test(unittest.TestCase): """ Make sure that the coregionalize kernel work with and without cython enabled