Better error handling

allenai · Dec 2, 2024 · fe04db8 · fe04db8
1 parent 35502bc
commit fe04db8
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/pdelfin/beakerpipeline.py b/pdelfin/beakerpipeline.py
@@ -517,9 +517,10 @@ async def sglang_server_host(args, semaphore):
         logger.warning("SGLang server task ended")
         retry += 1
 
-    logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
-    sys.exit(1)
-
+    if retry >= MAX_RETRIES:
+        logger.error(f"Ended up restarting the sglang server more than {retry} times, cancelling")
+        sys.exit(1)
+
 
 async def sglang_server_ready():
     max_attempts = 300