Install cuml-cu12

http://b/381287748
Kaggle · Dec 20, 2024 · c3342b8 · c3342b8
1 parent 5452732
commit c3342b8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/kaggle_requirements.txt b/kaggle_requirements.txt
@@ -23,6 +23,7 @@ catboost
 category-encoders
 cesium
 comm
+cuml-cu12
 cytoolz
 dask-expr
 # Older versions of datasets fail with "Loading a dataset cached in a LocalFileSystem is not supported"

diff --git a/tests/test_cuml.py b/tests/test_cuml.py
@@ -6,7 +6,6 @@
 class TestCuml(unittest.TestCase):
     @gpu_test
     @p100_exempt # b/342143152: cuML(>=24.4v) is inompatible with p100 GPUs.
-    @unittest.skip("b/381287748 cuML is not installed in Colab.")
     def test_pca_fit_transform(self):
         import unittest
         import numpy as np