feat(utils): Make suppress_outputs also disable logging

microsoft · Sep 11, 2023 · cda74ac · cda74ac
1 parent 58a2b51
commit cda74ac
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/syntheseus/reaction_prediction/utils/misc.py b/syntheseus/reaction_prediction/utils/misc.py
@@ -1,3 +1,4 @@
+import logging
 import multiprocessing
 import os
 import random
@@ -33,7 +34,9 @@ def suppress_outputs():
     """Suppress messages written to both stdout and stderr."""
     with open(devnull, "w") as fnull:
         with redirect_stderr(fnull), redirect_stdout(fnull):
+            logging.disable(logging.CRITICAL)
             yield
+            logging.disable(logging.NOTSET)
 
 
 def dictify(data: Any) -> Any: