Narrow the warning filters to be more specific, ensuring we aren't ac…

…cidentally ignoring other warnings
nv-morpheus · Dec 29, 2023 · f1dcd44 · f1dcd44
1 parent da692e8
commit f1dcd44
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/morpheus/utils/column_info.py b/morpheus/utils/column_info.py
@@ -23,7 +23,7 @@
 
 with warnings.catch_warnings():
     # Ignore warning regarding tensorflow not being installed
-    warnings.simplefilter("ignore")
+    warnings.filterwarnings("ignore", message=".*No module named 'tensorflow'", category=UserWarning)
     import nvtabular as nvt
 
 import pandas as pd

diff --git a/morpheus/utils/schema_transforms.py b/morpheus/utils/schema_transforms.py
@@ -29,7 +29,7 @@
 
 with warnings.catch_warnings():
     # Ignore warning regarding tensorflow not being installed
-    warnings.simplefilter("ignore")
+    warnings.filterwarnings("ignore", message=".*No module named 'tensorflow'", category=UserWarning)
     import nvtabular as nvt
 
     if os.environ.get("MORPHEUS_IN_SPHINX_BUILD") is None: