Disable all warnings.

xiejw · Jan 23, 2020 · e9ebe9f · e9ebe9f
1 parent be0708c
commit e9ebe9f
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 2 deletions.
diff --git a/connect_4/Makefile b/connect_4/Makefile
@@ -1,7 +1,7 @@
 default: run
 
 run:
-	PYTHONPATH=. python3 cmd/play_with_human.py
+	TF_CPP_MIN_LOG_LEVEL=3 PYTHONPATH=. python3 cmd/play_with_human.py
 
 train:
 	PYTHONPATH=. python3 cmd/train.py

diff --git a/connect_4/cmd/play_with_human.py b/connect_4/cmd/play_with_human.py
@@ -1,3 +1,26 @@
+# Disable all warnings from TensorFlow (Python side).
+#
+# - The order to disable warnings matters.
+# - For CPP side, set # TF_CPP_MIN_LOG_LEVEL=3.
+
+import warnings
+with warnings.catch_warnings():
+    warnings.filterwarnings("ignore",category=FutureWarning)
+
+    from tensorflow.python.util import deprecation
+    deprecation._PRINT_DEPRECATION_WARNINGS = False
+
+    import tensorflow as tf
+
+    # In addition, disable warning logging.
+    tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)
+
+    from tensorflow import keras
+
+##########################
+# Real fun code starts.
+##########################
+
 import random
 
 from game import GameConfig

diff --git a/connect_4/docker/run.sh b/connect_4/docker/run.sh
@@ -1,3 +1,3 @@
 #!/bin/sh
 
-PYTHONPATH=. python3 cmd/play_with_human.py
+TF_CPP_MIN_LOG_LEVEL=3  PYTHONPATH=. python3 cmd/play_with_human.py