Workaround to suporess (some) import warnings from NumPy

Co-authored-by: Ben Mares <[email protected]>

Author: 2 people

pymc/__init__.py

@@ -16,8 +16,7 @@
 __version__ = "4.1.1"
 
 import logging
-import multiprocessing as mp
-import platform
+import sys
 
 _log = logging.getLogger("pymc")
 
@@ -28,6 +27,74 @@
         _log.addHandler(handler)
 
 
+def __suppress_aesara_import_warnings():
+    """This is a workaround to suppress nonlethal NumPy warnings.
+
+    Some more printouts remain. See https://github.com/numpy/numpy/issues/21942
+
+    Remove this once https://github.com/aesara-devs/aesara/pull/980 is merged.
+    """
+    # We need to catch warnings as in some cases NumPy prints
+    # stuff that we don't want the user to see.
+    import io
+    import warnings
+
+    from contextlib import redirect_stderr, redirect_stdout
+
+    import numpy.distutils.system_info
+
+    class NumpyCompatibleStdoutStringIO(io.StringIO):
+        """Used as a temporary replacement of sys.stdout to capture Numpy's output.
+        We want to simply use io.StringIO, but this doesn't work because
+        Numpy expects the .encoding attribute to be a string. For io.StringIO,
+        this attribute is set to None and cannot be modified, hence the need for
+        this subclass.
+        (See forward_bytes_to_stdout in numpy.distutils.exec_command.)
+        """
+
+        encoding = sys.stdout.encoding
+
+    # Known executables which trigger false-positive warnings when not found.
+    # Ref: <https://github.com/conda-forge/aesara-feedstock/issues/54>
+    executables = ["g77", "f77", "ifort", "ifl", "f90", "DF", "efl"]
+
+    # The Numpy function which defines blas_info emits false-positive warnings.
+    # In what follows we capture these warnings and ignore them.
+    with warnings.catch_warnings(record=True):
+        # The warnings about missing executables don't use Python's "warnings"
+        # mechanism, and thus are not filtered by the catch_warnings context
+        # above. On Linux the warnings are printed to stderr, but on Windows
+        # they are printed to stdout. Thus we capture and filter both stdout
+        # and stderr.
+        stdout_sio, stderr_sio = NumpyCompatibleStdoutStringIO(), io.StringIO()
+        with redirect_stdout(stdout_sio), redirect_stderr(stderr_sio):
+            numpy.distutils.system_info.get_info("blas_opt")
+
+        # Print any unfiltered messages to stdout and stderr.
+        # (We hope that, beyond what we filter out, there should have been
+        # no messages printed to stdout or stderr. In case there were messages,
+        # we want to print them for the user to see. In what follows, we print
+        # the stdout messages followed by the stderr messages. This means that
+        # messages will be printed in the wrong order in the case that
+        # there is output to both stdout and stderr, and the stderr output
+        # doesn't come at the end.)
+        for captured_buffer, print_destination in (
+            (stdout_sio, sys.stdout),
+            (stderr_sio, sys.stderr),
+        ):
+            raw_lines = captured_buffer.getvalue().splitlines()
+            filtered_lines = [
+                line
+                for line in raw_lines
+                # Keep a line when none of the warnings are found within
+                # (equiv. when all of the warnings are not found within).
+                if all(f"Could not locate executable {exe}" not in line for exe in executables)
+            ]
+            for line in filtered_lines:
+                print(line, file=print_destination)
+    return
+
+
 def __set_compiler_flags():
     # Workarounds for Aesara compiler problems on various platforms
     import aesara
@@ -47,6 +114,8 @@ def __set_compiler_flags():
     aesara.config.gcc__cxxflags = augmented
 
 
+if sys.platform == "win32":
+    __suppress_aesara_import_warnings()
 __set_compiler_flags()
 
 from pymc import gp, ode, sampling