From 30a7b77f0edf549273c9c837d0c2db8e549cb4fd Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Mon, 19 Aug 2019 10:55:22 +0300 Subject: [PATCH] stage: run: don't use unassigned old_handler Fix #2406 --- dvc/stage.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/dvc/stage.py b/dvc/stage.py index e93d5ab92..373de499e 100644 --- a/dvc/stage.py +++ b/dvc/stage.py @@ -775,6 +775,7 @@ class Stage(object): main_thread = isinstance( threading.current_thread(), threading._MainThread ) + old_handler = None p = None try: @@ -790,7 +791,7 @@ class Stage(object): old_handler = signal.signal(signal.SIGINT, signal.SIG_IGN) p.communicate() finally: - if main_thread: + if old_handler: signal.signal(signal.SIGINT, old_handler) if (p is None) or (p.returncode != 0):