diff --git a/tests/integration/test_lists/waives.txt b/tests/integration/test_lists/waives.txt index 8e72d795b795..97e515fe7559 100644 --- a/tests/integration/test_lists/waives.txt +++ b/tests/integration/test_lists/waives.txt @@ -465,7 +465,6 @@ unittest/auto_deploy/multigpu/custom_ops/test_ad_dist_strategies.py::test_allred unittest/auto_deploy/multigpu/custom_ops/test_ad_dist_strategies.py::test_allreduce_strategies[NCCL] SKIP (https://nvbugs/6403920) unittest/auto_deploy/multigpu/custom_ops/test_ad_dist_strategies.py::test_allreduce_strategies[SYMM_MEM] SKIP (https://nvbugs/6403920) unittest/bindings/test_executor_bindings.py SKIP (TRTLLM-13781: legacy TensorRT examples removed; tests to be removed in follow-up PR3) -unittest/bindings/test_transfer_agent_bindings.py::TestNixlFunctionalTransfer::test_nixl_wait_in_progress_on_zero_timeout SKIP (https://nvbugs/6260897) unittest/disaggregated/test_kv_transfer.py::test_transfer_worker_v2[tp4_pp1_to_tp2_pp2] SKIP (https://nvbugs/6426834) unittest/executor/test_rpc.py::TestRpcCorrectness::test_incremental_task_async SKIP (https://nvbugs/5741476) unittest/executor/test_rpc_proxy.py SKIP (https://nvbugs/5605741) diff --git a/tests/unittest/bindings/test_transfer_agent_bindings.py b/tests/unittest/bindings/test_transfer_agent_bindings.py index ab1bf158348c..d6c9d5f440c0 100644 --- a/tests/unittest/bindings/test_transfer_agent_bindings.py +++ b/tests/unittest/bindings/test_transfer_agent_bindings.py @@ -585,8 +585,11 @@ def test_nixl_wait_in_progress_on_zero_timeout(self): status = agent_a.submit_transfer_requests(request) # With timeout_ms=0, wait checks status once and returns immediately. + # Either IN_PROGRESS (transfer still running) or SUCCESS (transfer + # already completed by the progress thread on a fast loopback path) + # is a valid non-blocking observation; FAILURE would indicate a bug. result = status.wait(timeout_ms=0) - assert result == tab.TransferState.IN_PROGRESS + assert result in (tab.TransferState.IN_PROGRESS, tab.TransferState.SUCCESS) # Wait for the transfer to actually finish before cleanup final_result = status.wait(timeout_ms=5000)