From 660cb4668476d3fbc773262d65eb9351dbb702cf Mon Sep 17 00:00:00 2001 From: Yan Chunwei <328693+Superjomn@users.noreply.github.com> Date: Thu, 4 Sep 2025 10:10:43 +0800 Subject: [PATCH] [https://nvbugs/5351244][fix] test_mpi_session (#7501) Signed-off-by: Yan Chunwei <328693+Superjomn@users.noreply.github.com> --- tests/unittest/llmapi/test_mpi_session.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/unittest/llmapi/test_mpi_session.py b/tests/unittest/llmapi/test_mpi_session.py index 484caf7381e1..bedce258c26d 100644 --- a/tests/unittest/llmapi/test_mpi_session.py +++ b/tests/unittest/llmapi/test_mpi_session.py @@ -54,11 +54,13 @@ def run_client(server_addr, values_to_process): return f"Error in client: {str(e)}" -@pytest.mark.skip(reason="https://nvbugs/5351244") @pytest.mark.parametrize("task_type", ["submit", "submit_sync"]) def test_remote_mpi_session(task_type: Literal["submit", "submit_sync"]): """Test RemoteMpiPoolSessionClient and RemoteMpiPoolSessionServer interaction""" - command = ["bash", "_test_remote_mpi_session.sh", task_type] + cur_dir = os.path.dirname(os.path.abspath(__file__)) + test_file = os.path.join(cur_dir, "_test_remote_mpi_session.sh") + assert os.path.exists(test_file), f"Test file {test_file} does not exist" + command = ["bash", test_file, task_type] print(' '.join(command)) with Popen(command,