Suppress deprecation warnings on tensorflow

HKU-BAL · Nov 10, 2019 · 03faea8 · 03faea8
1 parent 0729b4e
commit 03faea8
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/clair/model.py b/clair/model.py
@@ -1,6 +1,9 @@
 import warnings
 with warnings.catch_warnings():
-    warnings.filterwarnings("ignore",category=FutureWarning)
+    warnings.filterwarnings('ignore', category=DeprecationWarning)
+    warnings.filterwarnings("ignore", category=FutureWarning)
+    from tensorflow.python.util import deprecation
+    deprecation._PRINT_DEPRECATION_WARNINGS = False
     import tensorflow as tf
     from tensorflow.python.client import device_lib
     from tensorflow.python.ops import array_ops

diff --git a/clair/selu.py b/clair/selu.py
@@ -3,7 +3,10 @@
 '''
 import warnings
 with warnings.catch_warnings():
-    warnings.filterwarnings("ignore",category=FutureWarning)
+    warnings.filterwarnings('ignore', category=DeprecationWarning)
+    warnings.filterwarnings("ignore", category=FutureWarning)
+    from tensorflow.python.util import deprecation
+    deprecation._PRINT_DEPRECATION_WARNINGS = False
     import tensorflow as tf
     from tensorflow.contrib import layers
     from tensorflow.python.framework import ops