bug: timeout error

microsoft · Feb 6, 2025 · 2662671 · 2662671
1 parent 707b793
commit 2662671
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flaml/automl/model.py b/flaml/automl/model.py
@@ -93,7 +93,7 @@ def limit_resource(memory_limit, time_limit):
     if time_limit is not None:
         try:
             signal.signal(signal.SIGALRM, TimeoutHandler)
-            signal.alarm(int(time_limit) or 1)
+            signal.alarm(int(time_limit) + 1 or 1)
             main_thread = True
         except ValueError:
             pass