[GB200] In test_ultraserver, fix assert_imex_nodes_config_is_correct.

gmarciani · gmarciani · commit 4f41f341de70 · 2025-09-19T18:12:22.000-04:00
diff --git a/tests/integration-tests/tests/ultraserver/test_gb200.py b/tests/integration-tests/tests/ultraserver/test_gb200.py
@@ -207,7 +207,7 @@ def _check_imex_healthy():
             f"Private IP addresses for nodes in queue {queue} and compute resource {compute_resource}: " f"{ips}"
         )
 
-        assert_imex_nodes_config_is_correct(rce, queue, compute_resource, ips)
+        assert_imex_nodes_config_is_correct(cluster, queue, compute_resource, ips)
         assert_imex_status(rce, job_id, ips, service_status="UP", node_status="READY", connection_status="CONNECTED")
         assert_no_errors_in_logs(cluster, queue, compute_resource)
 

Original file line number	Diff line number	Diff line change
`@@ -207,7 +207,7 @@ def _check_imex_healthy():`
`207`	`207`	`f"Private IP addresses for nodes in queue {queue} and compute resource {compute_resource}: " f"{ips}"`
`208`	`208`	`)`
`209`	`209`
`210`		`- assert_imex_nodes_config_is_correct(rce, queue, compute_resource, ips)`
	`210`	`+ assert_imex_nodes_config_is_correct(cluster, queue, compute_resource, ips)`
`211`	`211`	`assert_imex_status(rce, job_id, ips, service_status="UP", node_status="READY", connection_status="CONNECTED")`
`212`	`212`	`assert_no_errors_in_logs(cluster, queue, compute_resource)`
`213`	`213`