Skip to content

Commit b6c3cb0

Browse files
committed
[GB200] Address linter findings.
1 parent 0cdccc5 commit b6c3cb0

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

tests/integration-tests/tests/ultraserver/test_gb200.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -55,13 +55,19 @@ def submit_job_imex_status(rce: RemoteCommandExecutor, queue: str, max_nodes: in
5555

5656
def assert_imex_nodes_config_is_correct(cluster: Cluster, queue: str, compute_resource: str, expected_ips: list):
5757
for compute_node_ip in cluster.get_compute_nodes_private_ip(queue, compute_resource):
58-
logging.info(f"Checking IMEX nodes config for compute node {compute_node_ip} contains the expected nodes: {expected_ips}")
58+
logging.info(
59+
f"Checking IMEX nodes config for compute node {compute_node_ip} contains the expected nodes: {expected_ips}"
60+
)
5961
rce = RemoteCommandExecutor(cluster, compute_node_ip=compute_node_ip)
6062
imex_config_content = read_remote_file(rce, "/etc/nvidia-imex/nodes_config.cfg")
61-
imex_config_content_clean = [line for line in imex_config_content.split("\n") if not line.strip().startswith("#")]
63+
imex_config_content_clean = [
64+
line for line in imex_config_content.split("\n") if not line.strip().startswith("#")
65+
]
6266
actual_ips = [ip.strip() for ip in imex_config_content_clean]
6367
assert_that(actual_ips).contains_only(*expected_ips)
64-
logging.info(f"IMEX nodes config for compute node {compute_node_ip} contains the expected nodes: {expected_ips}")
68+
logging.info(
69+
f"IMEX nodes config for compute node {compute_node_ip} contains the expected nodes: {expected_ips}"
70+
)
6571

6672

6773
def assert_no_errors_in_logs(cluster: Cluster, queue: str, compute_resource: str):

0 commit comments

Comments
 (0)