Unverified Commit ce50305e authored by Aymeric Augustin's avatar Aymeric Augustin Committed by GitHub
Browse files

Merge pull request #2270 from aaugustin/remove-python-2

Remove support for Python 2
parents b6ea0f43 1a948d70
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
"""PyTorch RoBERTa model. """ """PyTorch RoBERTa model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
# limitations under the License. # limitations under the License.
""" PyTorch T5 model. """ """ PyTorch T5 model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import copy import copy
import itertools import itertools
......
...@@ -14,10 +14,9 @@ ...@@ -14,10 +14,9 @@
# See the License for the specific language governing permissions and # See the License for the specific language governing permissions and
# limitations under the License. # limitations under the License.
""" TF 2.0 ALBERT model. """ """ TF 2.0 ALBERT model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import sys
import tensorflow as tf import tensorflow as tf
...@@ -311,9 +310,7 @@ class TFAlbertLayer(tf.keras.layers.Layer): ...@@ -311,9 +310,7 @@ class TFAlbertLayer(tf.keras.layers.Layer):
config.intermediate_size, kernel_initializer=get_initializer(config.initializer_range), name="ffn" config.intermediate_size, kernel_initializer=get_initializer(config.initializer_range), name="ffn"
) )
if isinstance(config.hidden_act, str) or ( if isinstance(config.hidden_act, str):
sys.version_info[0] == 2 and isinstance(config.hidden_act, unicode) # noqa: F821
):
self.activation = ACT2FN[config.hidden_act] self.activation = ACT2FN[config.hidden_act]
else: else:
self.activation = config.hidden_act self.activation = config.hidden_act
...@@ -454,9 +451,7 @@ class TFAlbertMLMHead(tf.keras.layers.Layer): ...@@ -454,9 +451,7 @@ class TFAlbertMLMHead(tf.keras.layers.Layer):
self.dense = tf.keras.layers.Dense( self.dense = tf.keras.layers.Dense(
config.embedding_size, kernel_initializer=get_initializer(config.initializer_range), name="dense" config.embedding_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"
) )
if isinstance(config.hidden_act, str) or ( if isinstance(config.hidden_act, str):
sys.version_info[0] == 2 and isinstance(config.hidden_act, unicode) # noqa: F821
):
self.activation = ACT2FN[config.hidden_act] self.activation = ACT2FN[config.hidden_act]
else: else:
self.activation = config.hidden_act self.activation = config.hidden_act
......
...@@ -14,7 +14,6 @@ ...@@ -14,7 +14,6 @@
# limitations under the License. # limitations under the License.
""" Auto Model class. """ """ Auto Model class. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,10 +15,8 @@ ...@@ -15,10 +15,8 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 BERT model. """ """ TF 2.0 BERT model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import sys
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
...@@ -311,9 +309,7 @@ class TFBertIntermediate(tf.keras.layers.Layer): ...@@ -311,9 +309,7 @@ class TFBertIntermediate(tf.keras.layers.Layer):
self.dense = tf.keras.layers.Dense( self.dense = tf.keras.layers.Dense(
config.intermediate_size, kernel_initializer=get_initializer(config.initializer_range), name="dense" config.intermediate_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"
) )
if isinstance(config.hidden_act, str) or ( if isinstance(config.hidden_act, str):
sys.version_info[0] == 2 and isinstance(config.hidden_act, unicode) # noqa: F821
):
self.intermediate_act_fn = ACT2FN[config.hidden_act] self.intermediate_act_fn = ACT2FN[config.hidden_act]
else: else:
self.intermediate_act_fn = config.hidden_act self.intermediate_act_fn = config.hidden_act
...@@ -418,9 +414,7 @@ class TFBertPredictionHeadTransform(tf.keras.layers.Layer): ...@@ -418,9 +414,7 @@ class TFBertPredictionHeadTransform(tf.keras.layers.Layer):
self.dense = tf.keras.layers.Dense( self.dense = tf.keras.layers.Dense(
config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense" config.hidden_size, kernel_initializer=get_initializer(config.initializer_range), name="dense"
) )
if isinstance(config.hidden_act, str) or ( if isinstance(config.hidden_act, str):
sys.version_info[0] == 2 and isinstance(config.hidden_act, unicode) # noqa: F821
):
self.transform_act_fn = ACT2FN[config.hidden_act] self.transform_act_fn = ACT2FN[config.hidden_act]
else: else:
self.transform_act_fn = config.hidden_act self.transform_act_fn = config.hidden_act
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 CTRL model.""" """ TF 2.0 CTRL model."""
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 DistilBERT model """ TF 2.0 DistilBERT model
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import math import math
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 OpenAI GPT-2 model. """ """ TF 2.0 OpenAI GPT-2 model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 OpenAI GPT model.""" """ TF 2.0 OpenAI GPT model."""
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" PyTorch - TF 2.0 general utilities.""" """ PyTorch - TF 2.0 general utilities."""
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import os import os
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 RoBERTa model. """ """ TF 2.0 RoBERTa model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 T5 model. """ """ TF 2.0 T5 model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import copy import copy
import itertools import itertools
......
...@@ -16,7 +16,6 @@ ...@@ -16,7 +16,6 @@
""" TF 2.0 Transformer XL model. """ TF 2.0 Transformer XL model.
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
"""TF general model utils.""" """TF general model utils."""
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import os import os
......
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 XLM model. """ TF 2.0 XLM model.
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import itertools import itertools
import logging import logging
......
...@@ -15,10 +15,9 @@ ...@@ -15,10 +15,9 @@
# limitations under the License. # limitations under the License.
""" TF 2.0 XLNet model. """ TF 2.0 XLNet model.
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import sys
import numpy as np import numpy as np
import tensorflow as tf import tensorflow as tf
...@@ -290,9 +289,7 @@ class TFXLNetFeedForward(tf.keras.layers.Layer): ...@@ -290,9 +289,7 @@ class TFXLNetFeedForward(tf.keras.layers.Layer):
config.d_model, kernel_initializer=get_initializer(config.initializer_range), name="layer_2" config.d_model, kernel_initializer=get_initializer(config.initializer_range), name="layer_2"
) )
self.dropout = tf.keras.layers.Dropout(config.dropout) self.dropout = tf.keras.layers.Dropout(config.dropout)
if isinstance(config.ff_activation, str) or ( if isinstance(config.ff_activation, str):
sys.version_info[0] == 2 and isinstance(config.ff_activation, unicode) # noqa: F821
):
self.activation_function = ACT2FN[config.ff_activation] self.activation_function = ACT2FN[config.ff_activation]
else: else:
self.activation_function = config.ff_activation self.activation_function = config.ff_activation
......
...@@ -18,7 +18,6 @@ ...@@ -18,7 +18,6 @@
In particular https://github.com/kimiyoung/transformer-xl/blob/master/pytorch/mem_transformer.py In particular https://github.com/kimiyoung/transformer-xl/blob/master/pytorch/mem_transformer.py
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
"""PyTorch BERT model.""" """PyTorch BERT model."""
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
import os import os
......
...@@ -14,7 +14,7 @@ ...@@ -14,7 +14,7 @@
# limitations under the License. # limitations under the License.
""" PyTorch XLM model. """ PyTorch XLM model.
""" """
from __future__ import absolute_import, division, print_function, unicode_literals
import itertools import itertools
import logging import logging
......
...@@ -15,7 +15,6 @@ ...@@ -15,7 +15,6 @@
# limitations under the License. # limitations under the License.
"""PyTorch XLM-RoBERTa model. """ """PyTorch XLM-RoBERTa model. """
from __future__ import absolute_import, division, print_function, unicode_literals
import logging import logging
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment