diff --git a/tests/unit/legate/driver/test_command.py b/tests/unit/legate/driver/test_command.py index d278ee1c0..be74c755d 100644 --- a/tests/unit/legate/driver/test_command.py +++ b/tests/unit/legate/driver/test_command.py @@ -1228,7 +1228,10 @@ def test_utility_n_multi_rank_with_launcher( result = m.cmd_bgwork(config, system, launcher) - assert result == ("-ll:bgwork", value) + if "ucx" in install_info.networks: + assert result == ("-ll:bgwork", value, "-ll:bgworkpin", "1") + else: + assert result == ("-ll:bgwork", value) @pytest.mark.parametrize("launch", ("mpirun", "jsrun", "srun")) @pytest.mark.parametrize("value", ("2", "3", "10"))