Internal change

PiperOrigin-RevId: 361957289

official/nlp/xlnet/run_classifier.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """XLNet classification finetuning runner in tf2.0."""
 
 import functools

official/nlp/xlnet/run_pretrain.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """XLNet pretraining runner in tf2.0."""
 
 import functools

official/nlp/xlnet/run_squad.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """XLNet SQUAD finetuning runner in tf2.0."""
 
 import functools

official/nlp/xlnet/squad_utils.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 # coding=utf-8
 """Utilities used in SQUAD task."""
 from __future__ import absolute_import
@@ -656,7 +656,7 @@ def convert_examples_to_features(examples, sp_model, max_seq_length, doc_stride,
       assert tok_start_position <= tok_end_position
 
     def _piece_to_id(x):
-      if six.PY2 and isinstance(x, unicode):
+      if six.PY2 and isinstance(x, unicode):  # pylint: disable=undefined-variable
         x = x.encode("utf-8")
       return sp_model.PieceToId(x)
 

official/nlp/xlnet/training_utils.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,22 +11,21 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """XLNet training utils."""
 
 import os
 import re
+from typing import Any, Callable, Dict, Optional, Text
 
 from absl import logging
-
-# pytype: disable=attribute-error
-# pylint: disable=g-bare-generic,unused-import
 import tensorflow as tf
-from typing import Any, Callable, Dict, Text, Optional
 
 from official.nlp.bert import model_training_utils
 from official.nlp.xlnet import data_utils
-from official.nlp.xlnet import xlnet_modeling as modeling
+
+# pytype: disable=attribute-error
+# pylint: disable=g-bare-generic,unused-import
 
 _MIN_SUMMARY_STEPS = 10
 

official/nlp/xlnet/xlnet_config.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """Utility functions used in XLNet model."""
 
 import json

official/nlp/xlnet/xlnet_modeling.py

-# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
+# Copyright 2021 The TensorFlow Authors. All Rights Reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-# ==============================================================================
+
 """Keras layers of XLNet model in TF 2.0."""
 
 import copy