diff --git a/rl_coach/__init__.py b/rl_coach/__init__.py index e69de29..3d03889 100644 --- a/rl_coach/__init__.py +++ b/rl_coach/__init__.py @@ -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)