suppress/restore stdout in jupyter

Author: jcschaff

pyvcell/vcml/__init__.py

@@ -26,21 +26,21 @@
     field_data_refs,
     load_antimony_file,
     load_antimony_str,
-    load_sbml_url,
     load_sbml_file,
     load_sbml_str,
-    load_vcml_url,
+    load_sbml_url,
     load_vcml_file,
     load_vcml_str,
+    load_vcml_url,
+    restore_stdout,
+    suppress_stdout,
     to_antimony_str,
     to_sbml_str,
     to_vcml_str,
     update_biomodel,
     write_antimony_file,
     write_sbml_file,
     write_vcml_file,
-    suppress_stdout,
-    restore_stdout,
 )
 from pyvcell.vcml.vcml_reader import VcmlReader
 from pyvcell.vcml.vcml_simulation import simulate
@@ -92,4 +92,6 @@
     "Field",
     "load_vcml_url",
     "load_sbml_url",
+    "suppress_stdout",
+    "restore_stdout",
 ]

pyvcell/vcml/utils.py

@@ -326,12 +326,14 @@ def refresh_biomodel(bio_model: Biomodel) -> Biomodel:
         return VcmlReader.biomodel_from_file(vcml_path=vcml_path)
 
 
-def suppress_stdout():
+def suppress_stdout() -> None:
     sys.stdout.flush()  # Ensure all Python-level stdout is flushed
     devnull = os.open(os.devnull, os.O_WRONLY)
     os.dup2(devnull, sys.stdout.fileno())
 
 
-def restore_stdout():
+def restore_stdout() -> None:
     sys.stdout.flush()
+    if sys.__stdout__ is None:
+        return
     os.dup2(sys.__stdout__.fileno(), sys.stdout.fileno())