Internal change

PiperOrigin-RevId: 300481917

official/nlp/modeling/models/bert_classifier.py

@@ -19,7 +19,7 @@ from __future__ import division
 # from __future__ import google_type_annotations
 from __future__ import print_function
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from official.nlp.modeling import networks
 

official/nlp/modeling/models/bert_classifier_test.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.nlp.modeling import networks

official/nlp/modeling/models/bert_pretrainer.py

@@ -20,7 +20,7 @@ from __future__ import division
 from __future__ import print_function
 
 import copy
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from official.nlp.modeling import networks
 

official/nlp/modeling/models/bert_pretrainer_test.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.nlp.modeling import networks

official/nlp/modeling/models/bert_span_labeler.py

@@ -19,7 +19,7 @@ from __future__ import division
 # from __future__ import google_type_annotations
 from __future__ import print_function
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from official.nlp.modeling import networks
 

official/nlp/modeling/models/bert_span_labeler_test.py

@@ -18,7 +18,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 from tensorflow.python.keras import keras_parameterized  # pylint: disable=g-direct-tensorflow-import
 from official.nlp.modeling import networks

official/nlp/optimization.py

@@ -20,7 +20,7 @@ from __future__ import print_function
 
 import re
 
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 
 class WarmUp(tf.keras.optimizers.schedules.LearningRateSchedule):

official/nlp/transformer/utils/tokenizer.py

@@ -26,7 +26,7 @@ import unicodedata
 import numpy as np
 import six
 from six.moves import xrange  # pylint: disable=redefined-builtin
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 PAD = "<pad>"
 PAD_ID = 0

official/nlp/transformer/utils/tokenizer_test.py

@@ -17,7 +17,7 @@
 import collections
 import tempfile
 
-import tensorflow.compat.v2 as tf  # pylint: disable=g-bad-import-order
+import tensorflow as tf
 
 from official.nlp.transformer.utils import tokenizer
 

official/nlp/xlnet/preprocess_classification_data.py

@@ -454,6 +454,4 @@ def main(_):
 
 
 if __name__ == "__main__":
-
-  tf.enable_v2_behavior()
   app.run(main)