Fix key pair file deletion logic during efa test (#5257)

* add vllm arm64 to release

* test efa

* try tf 2.19

* try tf 2.19

* Revert toml

* remove pr context

* remove pr context

Author: Jyothirmaikottu

test/dlc_tests/conftest.py

@@ -600,11 +600,9 @@ def ec2_instance(
     key_filename = test_utils.generate_ssh_keypair(ec2_client, ec2_key_name)
 
     def delete_ssh_keypair():
-        if test_utils.is_pr_context():
-            test_utils.destroy_ssh_keypair(ec2_client, key_filename)
-        else:
-            with open(KEYS_TO_DESTROY_FILE, "a") as destroy_keys:
-                destroy_keys.write(f"{key_filename}\n")
+        # test_utils.destroy_ssh_keypair(ec2_client, key_filename)
+        with open(KEYS_TO_DESTROY_FILE, "a") as destroy_keys:
+            destroy_keys.write(f"{key_filename}\n")
 
     request.addfinalizer(delete_ssh_keypair)
     print(f"EC2 instance AMI-ID: {ec2_instance_ami}")

test/testrunner.py

@@ -263,13 +263,16 @@ def delete_key_pairs(keyfile):
 
     :param keyfile: file with all of the keys to delete
     """
-    with open(keyfile) as key_destroy_file:
-        for key_file in key_destroy_file:
-            LOGGER.info(key_file)
-            ec2_client = boto3.client("ec2", config=Config(retries={"max_attempts": 10}))
-            if ".pem" in key_file:
-                _resp, keyname = destroy_ssh_keypair(ec2_client, key_file)
-                LOGGER.info(f"Deleted {keyname}")
+    try:
+        with open(keyfile) as key_destroy_file:
+            for key_file in key_destroy_file:
+                LOGGER.info(f"destroying {key_file} listed in {key_destroy_file}")
+                ec2_client = boto3.client("ec2", config=Config(retries={"max_attempts": 10}))
+                if ".pem" in key_file:
+                    _resp, keyname = destroy_ssh_keypair(ec2_client, key_file)
+                    LOGGER.info(f"Deleted {keyname}")
+    except Exception as e:
+        LOGGER.error(f"Failed to delete key pair with exception: {e}")
 
 
 def build_bai_docker_container():