Test out graceful timeout.

Author: belitskiy

.github/workflows/wait-for-connection-test.yaml

@@ -35,7 +35,7 @@ jobs:
       - uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # ratchet:actions/checkout@v4
         with:
           repository: google-ml-infra/actions
-          ref: polish_ssh_connection_action
+          ref: ssh_graceful_timeout
       # Halt for connection if workflow dispatch is told to or if it is a retry with the label halt_on_retry
       - name: Wait For Connection
         uses: ./actions/ci_connection/

actions/ci_connection/wait_for_connection.py

@@ -17,7 +17,6 @@
 import asyncio
 import logging
 import os
-import sys
 import time
 
 from get_labels import retrieve_labels