Timeout 60 seconds for wait_for_inputs_delete

tests/common.py

@@ -1,5 +1,6 @@
 import os
 import time
+from datetime import datetime
 from typing import List, Tuple
 
 from grpc._channel import _Rendezvous
@@ -95,7 +96,9 @@ def cleanup_inputs(stub, input_ids, metadata):
 
 def wait_for_inputs_delete(stub, input_ids, metadata):
     remaining_input_ids = list(input_ids)
-    while remaining_input_ids:
+    start = datetime.now()
+    timeout = 60
+    while remaining_input_ids and (datetime.now() - start).total_seconds() < timeout:
         for input_id in remaining_input_ids:
             get_input_response = stub.GetInput(
                 service_pb2.GetInputRequest(input_id=input_id), metadata=metadata
@@ -106,6 +109,8 @@ def wait_for_inputs_delete(stub, input_ids, metadata):
                 print(f"Waiting for input '{input_id}' to be deleted")
                 time.sleep(1)
                 break
+    if (datetime.now() - start).total_seconds() >= timeout:
+        raise Exception(f"Timeout after {timeout} seconds to delete inputs {remaining_input_ids}")
 
 
 def wait_for_model_trained(stub, metadata, model_id, model_version_id, user_app_id=None):