From 103d4477ebf42c4bcd9e6e1c8545bd9ff5249853 Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Thu, 24 Sep 2020 15:11:45 +0300
Subject: [PATCH] Disable NumPy and TF2 related warnings (#463)

---
 rl_coach/__init__.py | 7 +++++++
 1 file changed, 7 insertions(+)

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)