Avoid tf deprecation warning (#1148)

6 years ago · a2da181f01
parent 15dc61dd38
commit a2da181f01
1 changed files with 4 additions and 1 deletions
--- 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