Avoid tf deprecation warning

cvat-ai · Feb 17, 2020 · 18308f0 · 18308f0
1 parent 06529b9
commit 18308f0
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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