Update the tests so the cuda-requiring ones are skipped in CI

transientlunatic · Aug 23, 2024 · 58ac10d · 58ac10d
1 parent 651012d
commit 58ac10d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_inference_uncertain.py b/tests/test_inference_uncertain.py
@@ -68,7 +68,7 @@ def test_likelihood_maximum_at_true_value_mass_ratio(self):
 
         self.assertTrue(np.abs(mass_ratios[np.argmax(log_likes)] - 0.6) < 0.1)
 
-
+@unittest.skipIf(CUDA_NOT_AVAILABLE, "CUDA is not installed on this system")
 class Test_Likelihood_ZeroNoise_With_Uncertainty_PyTorch(unittest.TestCase):
     """
     Test likelihoods on a zero noise injection.