From 9cccd58681f2e67be3f3b257c8a76820d945a44b Mon Sep 17 00:00:00 2001 From: Sebastian Berg Date: Tue, 14 Nov 2023 21:35:43 +0100 Subject: [PATCH] Fix slicing in tests, although not sure it ever has enough workers --- python/raft-dask/raft_dask/test/test_comms.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python/raft-dask/raft_dask/test/test_comms.py b/python/raft-dask/raft_dask/test/test_comms.py index 467cc8fe0e..7c628d1888 100644 --- 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)