Mark test as CUDA-only

lancedb · Jul 3, 2024 · f446aa8 · f446aa8
1 parent 3634896
commit f446aa8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/python/tests/test_indices.py b/python/python/tests/test_indices.py
@@ -78,6 +78,7 @@ def test_num_partitions(tmpdir):
     assert ivf.num_partitions == 10
 
 
+@pytest.mark.cuda
 def test_assign_partitions(tmpdir):
     ds = gen_dataset(tmpdir)
     builder = IndicesBuilder(ds, "vectors")