Merge pull request #42 from jeffyjefflabs/patch-1

Fix crash in check for terminal ipython
minrk · Mar 30, 2021 · 10364c5 · 10364c5
2 parents 2b923ab + 2e43863
commit 10364c5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wurlitzer.py b/wurlitzer.py
@@ -365,7 +365,7 @@ def load_ipython_extension(ip):
     
     Use: %load_ext wurlitzer
     """
-    if not getattr(ip, 'kernel'):
+    if not getattr(ip, 'kernel', None):
         warnings.warn(
             "wurlitzer extension doesn't do anything in terminal IPython"
         )
@@ -379,7 +379,7 @@ def unload_ipython_extension(ip):
     
     Use: %unload_ext wurlitzer
     """
-    if not getattr(ip, 'kernel'):
+    if not getattr(ip, 'kernel', None):
         return  
     ip.events.unregister('pre_execute', sys_pipes_forever)
     ip.events.unregister('post_execute', stop_sys_pipes)