From fac49b7bbe6c13f63ae83041b0156e4986dc92ca Mon Sep 17 00:00:00 2001 From: Phuc Nguyen Date: Thu, 11 Jan 2024 14:09:05 +0000 Subject: [PATCH] fix tests --- tests/test_distributed.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_distributed.py b/tests/test_distributed.py index aa884e50..3e7cdc2f 100644 --- a/tests/test_distributed.py +++ b/tests/test_distributed.py @@ -18,7 +18,7 @@ def _test_init_parallel_context(parallel_context: ParallelContext): world_rank = dist.get_rank(parallel_context.world_pg) ranks3d = parallel_context.get_3d_ranks(world_rank) - assert type(ranks3d) and len(ranks3d) + assert isinstance(ranks3d, tuple) and len(ranks3d) assert isinstance(parallel_context.world_rank_matrix, np.ndarray) assert isinstance(parallel_context.world_ranks_to_pg, dict)