disable cuML test since its not on colab

Kaggle · Nov 27, 2024 · d916094 · d916094
1 parent 0104ac6
commit d916094
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_cuml.py b/tests/test_cuml.py
@@ -6,6 +6,7 @@
 class TestCuml(unittest.TestCase):
     @gpu_test
     @p100_exempt # b/342143152: cuML(>=24.4v) is inompatible with p100 GPUs.
+    @unittest.skip("cuML is not installed in Colab.")
     def test_pca_fit_transform(self):
         import unittest
         import numpy as np