Jupyter interrupt fix (wandb#1257)

farizrahman4u · web-flow · commit 678de3e7b9d3 · 2020-09-23T20:11:13.000+04:00
* fix

* flag-&gt;else:
diff --git a/wandb/sdk/wandb_run.py b/wandb/sdk/wandb_run.py
@@ -31,7 +31,15 @@
 from wandb.data_types import _datatypes_set_callback
 from wandb.errors import Error
 from wandb.interface.summary_record import SummaryRecord
-from wandb.lib import config_util, filenames, module, proto_util, redirect, sparkline
+from wandb.lib import (
+    config_util,
+    filenames,
+    ipython,
+    module,
+    proto_util,
+    redirect,
+    sparkline,
+)
 from wandb.util import add_import_hook, sentry_set_scope, to_forward_slash_path
 from wandb.viz import Visualize
 
@@ -1128,15 +1136,18 @@ def _atexit_cleanup(self, exit_code=None):
             self._on_finish()
         except KeyboardInterrupt:
             wandb.termerror("Control-C detected -- Run data was not synced")
-            os._exit(-1)
+            if ipython._get_python_type() == "python":
+                os._exit(-1)
         except Exception as e:
             self._console_stop()
             self._backend.cleanup()
             logger.error("Problem finishing run", exc_info=e)
             wandb.termerror("Problem finishing run")
             traceback.print_exception(*sys.exc_info())
-            os._exit(-1)
-        self._on_final()
+            if ipython._get_python_type() == "python":
+                os._exit(-1)
+        else:
+            self._on_final()
 
     def _console_start(self):
         logger.info("atexit reg")
diff --git a/wandb/sdk_py27/wandb_run.py b/wandb/sdk_py27/wandb_run.py
@@ -31,7 +31,15 @@
 from wandb.data_types import _datatypes_set_callback
 from wandb.errors import Error
 from wandb.interface.summary_record import SummaryRecord
-from wandb.lib import config_util, filenames, module, proto_util, redirect, sparkline
+from wandb.lib import (
+    config_util,
+    filenames,
+    ipython,
+    module,
+    proto_util,
+    redirect,
+    sparkline,
+)
 from wandb.util import add_import_hook, sentry_set_scope, to_forward_slash_path
 from wandb.viz import Visualize
 
@@ -1128,15 +1136,18 @@ def _atexit_cleanup(self, exit_code=None):
             self._on_finish()
         except KeyboardInterrupt:
             wandb.termerror("Control-C detected -- Run data was not synced")
-            os._exit(-1)
+            if ipython._get_python_type() == "python":
+                os._exit(-1)
         except Exception as e:
             self._console_stop()
             self._backend.cleanup()
             logger.error("Problem finishing run", exc_info=e)
             wandb.termerror("Problem finishing run")
             traceback.print_exception(*sys.exc_info())
-            os._exit(-1)
-        self._on_final()
+            if ipython._get_python_type() == "python":
+                os._exit(-1)
+        else:
+            self._on_final()
 
     def _console_start(self):
         logger.info("atexit reg")
