Skip to content

Commit c3342b8

Browse files
committed
Install cuml-cu12
http://b/381287748
1 parent 5452732 commit c3342b8

File tree

2 files changed

+1
-1
lines changed

2 files changed

+1
-1
lines changed

kaggle_requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ catboost
2323
category-encoders
2424
cesium
2525
comm
26+
cuml-cu12
2627
cytoolz
2728
dask-expr
2829
# Older versions of datasets fail with "Loading a dataset cached in a LocalFileSystem is not supported"

tests/test_cuml.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
class TestCuml(unittest.TestCase):
77
@gpu_test
88
@p100_exempt # b/342143152: cuML(>=24.4v) is inompatible with p100 GPUs.
9-
@unittest.skip("b/381287748 cuML is not installed in Colab.")
109
def test_pca_fit_transform(self):
1110
import unittest
1211
import numpy as np

0 commit comments

Comments
 (0)