fix: protect stdout/stderr restoration in `InProcessKernel._redirecte…

…d_io` (#1122)
ipython · Jun 8, 2023 · 1c7f626 · 1c7f626
1 parent 3455c29
commit 1c7f626
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ipykernel/inprocess/ipkernel.py b/ipykernel/inprocess/ipkernel.py
@@ -121,9 +121,11 @@ def _input_request(self, prompt, ident, parent, password=False):
     def _redirected_io(self):
         """Temporarily redirect IO to the kernel."""
         sys_stdout, sys_stderr = sys.stdout, sys.stderr
-        sys.stdout, sys.stderr = self.stdout, self.stderr
-        yield
-        sys.stdout, sys.stderr = sys_stdout, sys_stderr
+        try:
+            sys.stdout, sys.stderr = self.stdout, self.stderr
+            yield
+        finally:
+            sys.stdout, sys.stderr = sys_stdout, sys_stderr
 
     # ------ Trait change handlers --------------------------------------------