Skip to content

Commit 0133624

Browse files
committed
add gpu guard
1 parent 0219b1a commit 0133624

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/test_rapids.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,14 @@
11
import unittest
22

3-
import cudf
4-
from cuml.cluster import DBSCAN
5-
63
from common import gpu_test
74

85

96
class TestRapids(unittest.TestCase):
7+
@gpu_test
108
def test_dbscan(self):
9+
import cudf
10+
from cuml.cluster import DBSCAN
11+
1112
# Create and populate a GPU DataFrame
1213
gdf_float = cudf.DataFrame()
1314
gdf_float['0'] = [1.0, 2.0, 5.0]

0 commit comments

Comments
 (0)