From a2da181f0161186ba3808648b79719203dfd13c4 Mon Sep 17 00:00:00 2001
From: zhiltsov-max <zhiltsov.max35@gmail.com>
Date: Wed, 19 Feb 2020 23:40:03 +0300
Subject: [PATCH] Avoid tf deprecation warning (#1148)

---
 datumaro/datumaro/util/tf_util.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/datumaro/datumaro/util/tf_util.py b/datumaro/datumaro/util/tf_util.py
index 0d939a95..beaa4fc7 100644
--- a/datumaro/datumaro/util/tf_util.py
+++ b/datumaro/datumaro/util/tf_util.py
@@ -26,12 +26,15 @@ def import_tf():
         pass
 
     # Enable eager execution in early versions to unlock dataset operations
+    eager_enabled = False
     try:
         tf.compat.v1.enable_eager_execution()
+        eager_enabled = True
     except AttributeError:
         pass
     try:
-        tf.enable_eager_execution()
+        if not eager_enabled:
+            tf.enable_eager_execution()
     except AttributeError:
         pass