Guard against none soft timeout

simonsobs · Nov 15, 2024 · 4b9b837 · 4b9b837
1 parent 81a9e56
commit 4b9b837
Showing 1 changed file with 9 additions and 8 deletions.
diff --git a/librarian_background/rolling_deletion.py b/librarian_background/rolling_deletion.py
@@ -96,14 +96,15 @@ def core(self, session: Session):
         deleted = 0
         for instance in instances:
             # First, see if we've timed out.
-            if (datetime.now(timezone.utc) - core_begin) > self.soft_timeout:
-                logger.warning(
-                    "Ran out of time in deletion task! Only successfully deleted "
-                    "{n}/{m} instances; we will return later",
-                    n=deleted,
-                    m=len(instances),
-                )
-                return False
+            if self.soft_timeout is not None:
+                if (datetime.now(timezone.utc) - core_begin) > self.soft_timeout:
+                    logger.warning(
+                        "Ran out of time in deletion task! Only successfully deleted "
+                        "{n}/{m} instances; we will return later",
+                        n=deleted,
+                        m=len(instances),
+                    )
+                    return False
 
             # Check that we got what we wanted.
             try: