From bef75dea04d8ad89f77f2ae5e272ec7b3029ed8c Mon Sep 17 00:00:00 2001 From: Yan Chunwei <328693+Superjomn@users.noreply.github.com> Date: Wed, 3 Sep 2025 14:22:49 +0800 Subject: [PATCH] init 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 484caf7381e..bedce258c26 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,