From f446aa83422181d78ea7210cc475561b540e7868 Mon Sep 17 00:00:00 2001 From: Weston Pace Date: Wed, 3 Jul 2024 06:36:22 -0700 Subject: [PATCH] Mark test as CUDA-only --- python/python/tests/test_indices.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python/python/tests/test_indices.py b/python/python/tests/test_indices.py index 760a6ad7bc..80d7cc623b 100644 --- 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")