test methods

Author: Jyothirmaikottu

test/vllm/ec2/test_artifacts/test_ec2.py

@@ -379,13 +379,14 @@ def test_vllm_on_ec2(resources, image_uri):
 
             print("EFA tests completed successfully")
 
-            test_results["multi_node"] = run_multi_node_test(head_conn, worker_conn, image_uri)
-
             instance_id = list(ec2_connections.keys())[0]
             print(f"\n=== Running Single-Node Test on instance: {instance_id} ===")
             test_results["single_node"] = run_single_node_test(
                 ec2_connections[instance_id], image_uri
             )
+
+            test_results["multi_node"] = run_multi_node_test(head_conn, worker_conn, image_uri)
+
         else:
             print("\nSkipping multi-node test: insufficient instances")
 

test/vllm/ec2/utils/head_node_setup.sh

@@ -33,7 +33,5 @@ tmux new-session -d -s ray_head "docker run \
     ${IMAGE_URI} \
     /bin/bash -c 'ray start --head --block --port=6379'"
 
-docker ps -a 
-
 log "Head node started in container: ${CONTAINER_NAME}"