From 6014dc26d35db160f4e577d9f239175ff5dfe689 Mon Sep 17 00:00:00 2001 From: ilmarkov Date: Tue, 16 Dec 2025 15:02:53 +0000 Subject: [PATCH] Fix test Signed-off-by: ilmarkov --- tests/distributed/test_eplb_execute.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/distributed/test_eplb_execute.py b/tests/distributed/test_eplb_execute.py index dfb15311202e5..14ad148da6c45 100644 --- a/tests/distributed/test_eplb_execute.py +++ b/tests/distributed/test_eplb_execute.py @@ -312,7 +312,7 @@ def _test_async_transfer_layer_without_mtp_worker( is_received_locally=is_received_locally, recv_metadata=recv_metadata, new_indices=new_indices_cpu[layer_idx], - ep_group=ep_group, + ep_rank=ep_group.rank(), ) verify_expert_weights_after_shuffle(