[Integ-test] Fix get_compute_nodes_allocation

compute nodes list should be all running compute nodes. The previous function call was wrong and causing intermittent failures. I checked all usage of the function to make sure this commit is good. Signed-off-by: Hanwen <[email protected]>
aws · Sep 18, 2024 · c964a8b · c964a8b
1 parent ead1072
commit c964a8b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration-tests/tests/common/scaling_common.py b/tests/integration-tests/tests/common/scaling_common.py
@@ -226,7 +226,7 @@ def get_compute_nodes_allocation(
         stop_max_delay=max_monitoring_time,
     )
     def _watch_compute_nodes_allocation():
-        compute_nodes = len(scheduler_commands.get_unique_static_nodes())
+        compute_nodes = len(scheduler_commands.get_compute_nodes())
         ec2_capacity = get_compute_nodes_instance_count(stack_name, region)
         timestamp = time.time()