Explicitly replace "import tensorflow" with "tensorflow.compat.v1"

PiperOrigin-RevId: 286204718

meta_dataset/data/decoder.py

@@ -16,7 +16,7 @@
 # Lint as: python2, python3
 """Module responsible for decoding image/feature examples."""
 import gin.tf
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 @gin.configurable

meta_dataset/data/decoder_test.py

@@ -23,7 +23,7 @@
 from meta_dataset.data import decoder
 from meta_dataset.dataset_conversion import dataset_to_records
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class DecoderTest(tf.test.TestCase):

meta_dataset/data/imagenet_specification_test.py

@@ -24,7 +24,7 @@
 from meta_dataset.data import learning_spec
 import numpy as np
 from six.moves import range
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 DESIRED_TOY_NUM_VALID_CLASSES = 2
 DESIRED_TOY_NUM_TEST_CLASSES = 1

meta_dataset/data/pipeline.py

@@ -39,7 +39,7 @@
 from meta_dataset.data import reader
 from meta_dataset.data import sampling
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 def filter_dummy_examples(example_strings, class_ids):

meta_dataset/data/pipeline_test.py

@@ -26,7 +26,7 @@
 from meta_dataset.data.dataset_spec import DatasetSpecification
 from meta_dataset.dataset_conversion import dataset_to_records
 import numpy as np
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class PipelineTest(tf.test.TestCase):

meta_dataset/data/reader.py

@@ -35,7 +35,7 @@
 from meta_dataset import data
 import numpy as np
 from six.moves import range
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 # DUMMY_CLASS_ID will be used as the target of examples used for padding only.
 DUMMY_CLASS_ID = -1

meta_dataset/data/reader_test.py

@@ -31,7 +31,7 @@
 import numpy as np
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 # DatasetSpecification to use in tests
 DATASET_SPEC = DatasetSpecification(

meta_dataset/data/sampling_test.py

@@ -28,7 +28,7 @@
 import numpy as np
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 # DatasetSpecification to use in tests
 DATASET_SPEC = DatasetSpecification(

meta_dataset/learner.py

@@ -27,7 +27,7 @@
 import six
 from six.moves import range
 from six.moves import zip
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 MAX_WAY = 50  # The maximum number of classes we will see in any batch.
 

meta_dataset/trainer_test.py

@@ -25,7 +25,7 @@
 from meta_dataset import trainer
 from meta_dataset.data import config
 from meta_dataset.data import providers
-import tensorflow as tf
+import tensorflow.compat.v1 as tf
 
 
 class TrainerTest(tf.test.TestCase):