Update __init__.py

NetCDF changes stdout to the terminal. Here we try to reset it.
GeoscienceAustralia · May 11, 2018 · f283148 · f283148
1 parent 8a073b5
commit f283148
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/anuga/__init__.py b/anuga/__init__.py
@@ -50,6 +50,12 @@
         your python interpreter from there."""
         raise ImportError(msg)
 
+    #---------------------------------
+    # NetCDF changes stdout to terminal\
+    # Causes trouble when using jupyter
+    #---------------------------------
+    import sys
+    _stdout = sys.stdout    
 
     #---------------------------------
     # Setup the nose tester from numpy
@@ -331,9 +337,15 @@ def get_args():
 
     from anuga.config import g
     from anuga.config import velocity_protection
+
+    #--------------------------------------
+    # NetCDF changes stdout to the terminal
+    # This resets it
+    #--------------------------------------
+    reload(sys)
+    sys.stdout = _stdout
+
 
-
-