gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01:00
Code Issues Projects Releases Wiki Activity

Disable NumPy and TF2 related warnings (#463)

Browse Source
This commit is contained in:
Guy Jacob
2020-09-24 15:11:45 +03:00
committed by GitHub
parent c9738280fd
commit 103d4477eb
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
rl_coach/__init__.py
View File

@@ -0,0 +1,7 @@
# This gets rid of NumPy FutureWarnings that occur at TF import
import warnings
warnings.filterwarnings('ignore',category=FutureWarning)
# This gets rid of TF 2.0 related deprecation warnings
import tensorflow as tf
tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR)