Revert "Change how TF 2 is checked" (#7260)

This reverts commit 712f473e.

Revert "Change how TF 2 is checked" (#7260)
This reverts commit 712f473e.
2569fa9a · Toby Boyd · GitHub · 283de38b · 2569fa9a
Unverified Commit 2569fa9a authored Jul 19, 2019 by Toby Boyd Committed by GitHub Jul 19, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

official/utils/misc/keras_utils.py official/utils/misc/keras_utils.py +4 -2

No files found.
--- a/official/utils/misc/keras_utils.py
+++ b/official/utils/misc/keras_utils.py
@@ -204,5 +204,7 @@ def set_config_v2(enable_xla=False,

 def is_v2_0():
  """Returns true if using tf 2.0."""
-  from tensorflow.python import tf2
-  return tf2.enabled()
+  if hasattr(tf, 'contrib'):
+    return False
+  else:
+    return True