From 2b1403bd7bd84e312d26bf2fa6b57a6a8b7446a8 Mon Sep 17 00:00:00 2001
From: Altan Orhon <altan@uw.edu>
Date: Thu, 28 Sep 2023 14:47:37 -0700
Subject: [PATCH] Signal handling

---
 hyakvnc/__main__.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/hyakvnc/__main__.py b/hyakvnc/__main__.py
index 7dda050..dcee1b4 100644
--- a/hyakvnc/__main__.py
+++ b/hyakvnc/__main__.py
@@ -184,9 +184,7 @@ def get_session():
                 logger.debug(f"Could not get session info for job {job_id}: {e}")
             return None
 
-        sesh = repeat_until(
-            lambda x: get_session(), lambda x: x is not None, timeout=app_config.sbatch_post_timeout * 2
-        )
+        sesh = repeat_until(lambda: get_session(), lambda x: x is not None, timeout=app_config.sbatch_post_timeout * 2)
         if not sesh:
             logger.warning(f"No running VNC sessions found for job {job_id}. Canceling and exiting.")
             kill_self()