[GB200] In test_ultraserver, fix assert_imex_nodes_config_is_correct

gmarciani · gmarciani · commit 5eba4a61846e · 2025-09-19T18:20:15.000-04:00
diff --git a/tests/integration-tests/tests/ultraserver/test_gb200.py b/tests/integration-tests/tests/ultraserver/test_gb200.py
@@ -237,7 +237,7 @@ def assert_imex_not_configured(cluster: Cluster, queue: str, compute_resource: s
 
     job_id = submit_job_imex_status(rce, queue, max_nodes)
 
-    assert_imex_nodes_config_is_correct(rce, queue, compute_resource, FAKE_IPS)
+    assert_imex_nodes_config_is_correct(cluster, queue, compute_resource, FAKE_IPS)
     assert_imex_status(
         rce, job_id, FAKE_IPS, service_status="DOWN", node_status="UNAVAILABLE", connection_status="INVALID"
     )

Original file line number	Diff line number	Diff line change
`@@ -237,7 +237,7 @@ def assert_imex_not_configured(cluster: Cluster, queue: str, compute_resource: s`
`237`	`237`
`238`	`238`	`job_id = submit_job_imex_status(rce, queue, max_nodes)`
`239`	`239`
`240`		`- assert_imex_nodes_config_is_correct(rce, queue, compute_resource, FAKE_IPS)`
	`240`	`+ assert_imex_nodes_config_is_correct(cluster, queue, compute_resource, FAKE_IPS)`
`241`	`241`	`assert_imex_status(`
`242`	`242`	`rce, job_id, FAKE_IPS, service_status="DOWN", node_status="UNAVAILABLE", connection_status="INVALID"`
`243`	`243`	`)`