Skip to content

Commit 978ea9d

Browse files
committed
Update test's expectations
1 parent dfb11a8 commit 978ea9d

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

tests/json_gen_strategy/test_nccl_kubernetes_json_gen_strategy.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
# SPDX-FileCopyrightText: NVIDIA CORPORATION & AFFILIATES
2-
# Copyright (c) 2024-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright (c) 2024-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
# SPDX-License-Identifier: Apache-2.0
44
#
55
# Licensed under the Apache License, Version 2.0 (the "License");
@@ -141,7 +141,11 @@ def test_launcher_command_generation(self, test_run_with_extra_args: TestRun, k8
141141
assert "mpirun" in launcher_args
142142
assert f"-np {test_run_with_extra_args.nnodes * k8s_system.gpus_per_node}" in launcher_args
143143
assert "-bind-to none" in launcher_args
144-
assert "-mca plm_rsh_args '-o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null'" in launcher_args
144+
assert (
145+
f"-mca plm_rsh_args '-p {json_gen_strategy.ssh_port}"
146+
+ " -o StrictHostKeyChecking=no -o UserKnownHostsFile=/dev/null'"
147+
in launcher_args
148+
)
145149
assert nccl.cmd_args.subtest_name in launcher_args
146150
assert f"--nthreads {nccl.cmd_args.nthreads}" in launcher_args
147151
assert f"--ngpus {nccl.cmd_args.ngpus}" in launcher_args

0 commit comments

Comments
 (0)