Fix slicing in tests, although not sure it ever has enough workers

rapidsai · Nov 14, 2023 · fca015e · fca015e
1 parent ad0c0b6
commit fca015e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/python/raft-dask/raft_dask/test/test_comms.py b/python/raft-dask/raft_dask/test/test_comms.py
@@ -247,7 +247,9 @@ def test_collectives(client, func, root_location):
 
 
 @pytest.mark.nccl
-@pytest.mark.parametrize("subset", [-1, 1, slice(None, None, -2)])
+@pytest.mark.parametrize(
+    "subset", [slice(-1, None), slice(1), slice(None, None, -2)]
+)
 def test_comm_init_worker_subset(client, subset):
     # Basic test that initializing a subset of workers is fine
     cb = Comms(comms_p2p=True, verbose=True)