fix: run callback

phi-friday · Mar 1, 2024 · c504c23 · c504c23
1 parent da7887e
commit c504c23
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/timeout_executor/terminate.py b/src/timeout_executor/terminate.py
@@ -164,5 +164,7 @@ def terminate(process: subprocess.Popen[str], terminator: Terminator) -> None:
 
 
 def callback(callback_args: CallbackArgs, terminator: Terminator) -> None:
-    callback_args.process.wait()
-    terminator.run_callbacks(callback_args, terminator.func_name)
+    try:
+        callback_args.process.wait()
+    finally:
+        terminator.run_callbacks(callback_args, terminator.func_name)