import tensorflow.compat.v2 as tf

rickwierenga · rickwierenga · commit 4adb5224f3d8 · 2020-01-10T14:51:06.000Z
diff --git a/tensorflow_datasets/image/imagewang.py b/tensorflow_datasets/image/imagewang.py
@@ -21,7 +21,7 @@
 
 import os
 
-import tensorflow as tf
+import tensorflow.compat.v2 as tf
 import tensorflow_datasets.public_api as tfds
 
 # TODO(imagewang): BibTeX citation
