Skip tracebacks for keyboard interrupt

danijar · May 11, 2024 · 2682b5e · 2682b5e
1 parent 7685ecc
commit 2682b5e
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 3 deletions.
diff --git a/zerofun/__init__.py b/zerofun/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '2.0.5'
+__version__ = '2.0.6'
 
 import multiprocessing as mp
 try:

diff --git a/zerofun/process.py b/zerofun/process.py
@@ -96,6 +96,8 @@ def _wrapper(fn, name, args, lock, errqueue, inits):
       fn = cloudpickle.loads(fn)
       fn(*args)
       sys.exit(0)
+    except (SystemExit, KeyboardInterrupt):
+      pass
     except Exception as e:
       utils.warn_remote_error(e, name, lock)
       errqueue.put(e)

diff --git a/zerofun/thread.py b/zerofun/thread.py
@@ -72,8 +72,8 @@ def __repr__(self):
   def _wrapper(self, *args):
     try:
       self.fn(*args)
-    except SystemExit:
-      return
+    except (SystemExit, KeyboardInterrupt):
+      pass
     except Exception as e:
       utils.warn_remote_error(e, self.name)
       self._exitcode = 1