Py3 cleanup for nlp/

PiperOrigin-RevId: 339071563

official/nlp/albert/configs.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """The ALBERT configurations."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import six
 
 from official.nlp.bert import configs

official/nlp/albert/export_albert_tfhub.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """A script to export the ALBERT core model as a TF-Hub SavedModel."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 
 # Import libraries
 from absl import app

official/nlp/albert/export_albert_tfhub_test.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Tests official.nlp.albert.export_albert_tfhub."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 import numpy as np

official/nlp/albert/run_classifier.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """ALBERT classification finetuning runner in tf2.x."""
 
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 # Import libraries

official/nlp/albert/run_squad.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Run ALBERT on SQuAD 1.1 and SQuAD 2.0 in TF 2.x."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 import time

official/nlp/bert/bert_models.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """BERT models that are compatible with TF 2.0."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import gin
 import tensorflow as tf
 import tensorflow_hub as hub

official/nlp/bert/bert_models_test.py

@@ -12,9 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import tensorflow as tf
 

official/nlp/bert/configs.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """The main BERT model and related functions."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import copy
 import json
 

official/nlp/bert/export_tfhub.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """A script to export the BERT core model as a TF-Hub SavedModel."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 
 # Import libraries
 from absl import app

official/nlp/bert/export_tfhub_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Tests official.nlp.bert.export_tfhub."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 from absl.testing import parameterized

official/nlp/bert/input_pipeline.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """BERT model input pipelines."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 

official/nlp/bert/model_saving_utils.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """Utilities to save models."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import os
 
 from absl import logging

official/nlp/bert/model_training_utils.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """A light weight utilities to train NLP models."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 import tempfile

official/nlp/bert/model_training_utils_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Tests for official.modeling.training.model_training_utils."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 from absl import logging

official/nlp/bert/run_classifier.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """BERT classification or regression finetuning runner in TF 2.x."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import functools
 import json

official/nlp/bert/run_pretraining.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Run masked LM/next sentence pre-training for BERT in TF 2.x."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 # Import libraries
 from absl import app

official/nlp/bert/run_squad.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Run BERT on SQuAD 1.1 and SQuAD 2.0 in TF 2.x."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 import time

official/nlp/bert/run_squad_helper.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Library for running BERT family models on SQuAD 1.1/2.0 in TF 2.x."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import collections
 import json

official/nlp/bert/squad_evaluate_v1_1.py

@@ -22,10 +22,6 @@ Pranav Rajpurkar, Jian Zhang, Konstantin Lopyrev, Percy Liang
 https://nlp.stanford.edu/pubs/rajpurkar2016squad.pdf
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import re
 import string

official/nlp/bert/squad_evaluate_v2_0.py

@@ -22,10 +22,6 @@ This file is expected to map question ID's to the model's predicted probability
 that a question is unanswerable.
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import re
 import string

official/nlp/bert/tf1_checkpoint_converter_lib.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 r"""Convert checkpoints created by Estimator (tf1) to be Keras compatible."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import numpy as np
 import tensorflow.compat.v1 as tf  # TF 1.x

official/nlp/bert/tf2_encoder_checkpoint_converter.py

@@ -18,9 +18,6 @@ The conversion will yield an object-oriented checkpoint that can be used
 to restore a BertEncoder or BertPretrainerV2 object (see the `converted_model`
 FLAG below).
 """
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import os
 

official/nlp/bert/tokenization.py

@@ -19,10 +19,6 @@ The file is forked from:
 https://github.com/google-research/bert/blob/master/tokenization.py.
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import re
 import unicodedata

official/nlp/bert/tokenization_test.py

@@ -12,9 +12,6 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 # ==============================================================================
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import os
 import tempfile

official/nlp/data/classifier_data_lib.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """BERT library to process data for classification task."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import csv
 import importlib

official/nlp/data/create_finetuning_data.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """BERT finetuning task dataset generator."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import functools
 import json
 import os

official/nlp/data/create_pretraining_data.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Create masked LM/next sentence masked_lm TF examples for BERT."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import collections
 import itertools

official/nlp/data/squad_lib.py

@@ -13,12 +13,7 @@
 # limitations under the License.
 # ==============================================================================
 """Library to process data for SQuAD 1.1 and SQuAD 2.0."""
-
 # pylint: disable=g-bad-import-order
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import copy
 import json

official/nlp/data/squad_lib_sp.py

@@ -18,10 +18,6 @@ The file is forked from:
 
 https://github.com/google-research/ALBERT/blob/master/run_squad_sp.py
 """
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import collections
 import copy
 import json

official/nlp/modeling/networks/classification.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Classification and regression network."""
 # pylint: disable=g-classes-have-attributes
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 import collections
 import tensorflow as tf
 

official/nlp/modeling/networks/encoder_scaffold.py

@@ -15,11 +15,6 @@
 # ==============================================================================
 """Transformer-based text encoder network."""
 # pylint: disable=g-classes-have-attributes
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import inspect
 
 from absl import logging

official/nlp/modeling/networks/span_labeling.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Span labeling network."""
 # pylint: disable=g-classes-have-attributes
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 import collections
 import tensorflow as tf
 

official/nlp/nhnet/decoder.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """Transformer decoder that mimics a BERT encoder, to load BERT checkpoints."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import tensorflow as tf
 from official.modeling import tf_utils
 from official.nlp.modeling import layers

official/nlp/nhnet/decoder_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Tests for nlp.nhnet.decoder."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import numpy as np
 import tensorflow as tf
 from official.nlp.modeling import layers

official/nlp/nhnet/input_pipeline.py

@@ -14,11 +14,7 @@
 # ==============================================================================
 """Input pipelines."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
-import tensorflow.compat.v2 as tf
+import tensorflow as tf
 
 
 def decode_record(record, name_to_features):

official/nlp/nhnet/models.py

@@ -13,12 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """tf.keras Models for NHNet."""
-
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 from absl import logging
 import gin
 import tensorflow as tf

official/nlp/nhnet/optimizer.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """Optimizer and learning rate scheduler."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import tensorflow as tf
 
 from official.modeling.hyperparams import params_dict

official/nlp/nhnet/trainer.py

@@ -15,10 +15,6 @@
 # ==============================================================================
 """Run NHNet model training and eval."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 # Import libraries

official/nlp/nhnet/utils.py

@@ -13,11 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Utility helpers for Bert2Bert."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 from absl import logging
 import tensorflow as tf
 from typing import Optional, Text

official/nlp/transformer/attention_layer.py

@@ -13,11 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Implementation of multiheaded attention and self-attention layers."""
-
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import math
 
 import tensorflow as tf

official/nlp/transformer/compute_bleu.py

@@ -18,10 +18,6 @@ Source:
 https://github.com/tensorflow/tensor2tensor/blob/master/tensor2tensor/utils/bleu_hook.py
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import re
 import sys
 import unicodedata

official/nlp/transformer/data_download.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Download and preprocess WMT17 ende training and evaluation datasets."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 import random
 import tarfile

official/nlp/transformer/data_pipeline.py

@@ -47,10 +47,6 @@ Two things to note in the pipeline:
    in the order of the examples.
 """
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 
 from absl import logging

official/nlp/transformer/embedding_layer.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Implementation of embedding layer with shared weights."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 

official/nlp/transformer/ffn_layer.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Implementation of fully connected network."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 

official/nlp/transformer/metrics.py

@@ -22,9 +22,6 @@ Metrics:
  - ROUGE score. Source:
      https://github.com/tensorflow/tensor2tensor/blob/master/tensor2tensor/utils/rouge.py
 """
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import functools
 

official/nlp/transformer/misc.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Misc for Transformer."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 # pylint: disable=g-bad-import-order
 
 from absl import flags

official/nlp/transformer/model_utils.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Transformer model helper methods."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import math
 
 import numpy as np

official/nlp/transformer/model_utils_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Test Transformer model helper methods."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 from official.nlp.transformer import model_utils

official/nlp/transformer/optimizer.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Optimizer from addons and learning rate scheduler."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 

official/nlp/transformer/transformer.py

@@ -17,9 +17,6 @@
 Model paper: https://arxiv.org/pdf/1706.03762.pdf
 Transformer model code source: https://github.com/tensorflow/tensor2tensor
 """
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import tensorflow as tf
 from official.nlp.modeling.layers import position_embedding

official/nlp/transformer/transformer_layers_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Tests for layers in Transformer."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 from official.nlp.transformer import attention_layer

official/nlp/transformer/transformer_main.py

@@ -17,9 +17,6 @@
 See README for description of setting the training schedule and evaluating the
 BLEU score.
 """
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import os
 import tempfile

official/nlp/transformer/transformer_main_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Test Transformer model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import os
 import re
 import sys

official/nlp/transformer/transformer_test.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Test Transformer model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import tensorflow as tf
 
 from official.nlp.transformer import model_params

official/nlp/transformer/translate.py

@@ -14,10 +14,6 @@
 # ==============================================================================
 """Translate text or files using trained transformer model."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 # Import libraries
 from absl import logging
 import numpy as np

official/nlp/xlnet/common_flags.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Common flags used in XLNet model."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 
 from absl import flags
 

official/nlp/xlnet/data_utils.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """Utilities used for data preparation."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import collections
 import json
 import os

official/nlp/xlnet/optimization.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Functions and classes related to optimization (weight updates)."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 
 from absl import logging
 import tensorflow as tf

official/nlp/xlnet/preprocess_classification_data.py

@@ -13,9 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """Script to pre-process classification data into tfrecords."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import collections
 import csv

official/nlp/xlnet/preprocess_pretrain_data.py

@@ -15,10 +15,6 @@
 # ==============================================================================
 """Script to pre-process pre-training data into tfrecords."""
 
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
-
 import json
 import os
 import random

official/nlp/xlnet/preprocess_squad_data.py

@@ -14,9 +14,6 @@
 # limitations under the License.
 # ==============================================================================
 """Script to pre-process SQUAD data into tfrecords."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 
 import os
 import random

official/nlp/xlnet/preprocess_utils.py

@@ -14,9 +14,6 @@
 # limitations under the License.
 # ==============================================================================
 """Utilities for pre-processing."""
-from __future__ import absolute_import
-from __future__ import division
-from __future__ import print_function
 import unicodedata
 
 import six

official/nlp/xlnet/training_utils.py

@@ -13,10 +13,6 @@
 # limitations under the License.
 # ==============================================================================
 """XLNet training utils."""
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
 
 import os
 import re

official/nlp/xlnet/xlnet_config.py

@@ -14,11 +14,6 @@
 # ==============================================================================
 """Utility functions used in XLNet model."""
 
-from __future__ import absolute_import
-from __future__ import division
-# from __future__ import google_type_annotations
-from __future__ import print_function
-
 import json
 import os