Fix F401 flake8 warning (x88 / 116).

This change is mostly autogenerated with:

    $ python -m autoflake --in-place --recursive --remove-all-unused-imports --ignore-init-module-imports examples templates transformers utils hubconf.py setup.py

I made minor changes in the generated diff.
This commit is contained in:
Aymeric Augustin
2019-12-21 21:54:07 +01:00
parent 80327a13ea
commit 783a616999
52 changed files with 30 additions and 85 deletions

View File

@@ -19,14 +19,13 @@ import math
import os
import time
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.optim import AdamW
from torch.utils.data import BatchSampler, DataLoader, RandomSampler
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm, trange
from tqdm import tqdm
import psutil
from grouped_batch_sampler import GroupedBatchSampler, create_lengths_groups

View File

@@ -20,7 +20,7 @@ import argparse
import torch
from transformers import BertForMaskedLM, GPT2LMHeadModel, RobertaForMaskedLM
from transformers import GPT2LMHeadModel, RobertaForMaskedLM
if __name__ == "__main__":

View File

@@ -20,7 +20,7 @@ import argparse
import torch
from transformers import BertForMaskedLM, RobertaForMaskedLM
from transformers import BertForMaskedLM
if __name__ == "__main__":

View File

@@ -26,8 +26,7 @@ from datetime import datetime
import numpy as np
import torch
from torch.nn import CrossEntropyLoss, MSELoss
from torch.utils.data import DataLoader, SequentialSampler, Subset, TensorDataset
from torch.utils.data import DataLoader, SequentialSampler, Subset
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm

View File

@@ -26,7 +26,7 @@ import timeit
import numpy as np
import torch
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler, TensorDataset
from torch.utils.data import DataLoader, RandomSampler, SequentialSampler
from torch.utils.data.distributed import DistributedSampler
from tqdm import tqdm, trange

View File

@@ -1,5 +1,4 @@
# coding=utf-8
import _pickle as pickle
import collections
import datetime
import glob

View File

@@ -18,8 +18,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import logging
import six
from .configuration_utils import PretrainedConfig

View File

@@ -21,10 +21,8 @@
from __future__ import absolute_import, division, print_function, unicode_literals
import itertools
import logging
import numpy as np
import tensorflow as tf
from .configuration_xxx import XxxConfig

View File

@@ -20,7 +20,6 @@
from __future__ import absolute_import, division, print_function, unicode_literals
import itertools
import logging
import os
@@ -30,7 +29,7 @@ from torch.nn import CrossEntropyLoss, MSELoss
from .configuration_xxx import XxxConfig
from .file_utils import add_start_docstrings
from .modeling_utils import PreTrainedModel, prune_linear_layer
from .modeling_utils import PreTrainedModel
logger = logging.getLogger(__name__)

View File

@@ -24,7 +24,6 @@ from .utils import CACHE_DIR, require_tf, slow
if is_tf_available():
import tensorflow as tf
from transformers.modeling_tf_xxx import (
TFXxxModel,
TFXxxForMaskedLM,

View File

@@ -19,7 +19,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import collections
import logging
import os
import unicodedata
from io import open
from .tokenization_utils import PreTrainedTokenizer

View File

@@ -1,7 +1,6 @@
from argparse import ArgumentParser, Namespace
from logging import getLogger
from transformers import AutoModel, AutoTokenizer
from transformers.commands import BaseTransformersCLICommand

View File

@@ -18,8 +18,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import logging
import six
from .configuration_utils import PretrainedConfig

View File

@@ -20,8 +20,6 @@ import argparse
import logging
import os
import tensorflow as tf
from transformers import (
ALBERT_PRETRAINED_CONFIG_ARCHIVE_MAP,
BERT_PRETRAINED_CONFIG_ARCHIVE_MAP,

View File

@@ -20,7 +20,6 @@ import argparse
import logging
import pathlib
import numpy as np
import torch
from packaging import version

View File

@@ -16,9 +16,7 @@ import re
import string
from io import open
from tqdm import tqdm
from transformers.tokenization_bert import BasicTokenizer, whitespace_tokenize
from transformers.tokenization_bert import BasicTokenizer
logger = logging.getLogger(__name__)

View File

@@ -8,8 +8,8 @@ import numpy as np
from tqdm import tqdm
from ...file_utils import is_tf_available, is_torch_available
from ...tokenization_bert import BasicTokenizer, whitespace_tokenize
from .utils import DataProcessor, InputExample, InputFeatures
from ...tokenization_bert import whitespace_tokenize
from .utils import DataProcessor
if is_torch_available():

View File

@@ -21,7 +21,6 @@ from typing import List
import requests
import six
from requests.exceptions import HTTPError
from tqdm import tqdm

View File

@@ -32,7 +32,6 @@ from .configuration_auto import (
XLMRobertaConfig,
XLNetConfig,
)
from .file_utils import add_start_docstrings
from .modeling_albert import (
ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP,
AlbertForMaskedLM,
@@ -76,7 +75,6 @@ from .modeling_roberta import (
)
from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadModel
from .modeling_transfo_xl import TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP, TransfoXLLMHeadModel, TransfoXLModel
from .modeling_utils import PreTrainedModel, SequenceSummary
from .modeling_xlm import (
XLM_PRETRAINED_MODEL_ARCHIVE_MAP,
XLMForQuestionAnswering,

View File

@@ -23,11 +23,10 @@ import numpy as np
import torch
import torch.nn as nn
from torch.nn import CrossEntropyLoss
from torch.nn.parameter import Parameter
from .configuration_ctrl import CTRLConfig
from .file_utils import add_start_docstrings
from .modeling_utils import Conv1D, PreTrainedModel, SequenceSummary, prune_conv1d_layer
from .modeling_utils import Conv1D, PreTrainedModel
logger = logging.getLogger(__name__)

View File

@@ -19,7 +19,6 @@
from __future__ import absolute_import, division, print_function, unicode_literals
import copy
import itertools
import logging
import math

View File

@@ -21,7 +21,6 @@ import os
import torch
from torch import nn
from tqdm import trange
from .modeling_auto import AutoModel, AutoModelWithLMHead

View File

@@ -24,7 +24,6 @@ import os
import torch
import torch.nn as nn
from torch.nn import CrossEntropyLoss
from torch.nn.parameter import Parameter
from .configuration_gpt2 import GPT2Config
from .file_utils import add_start_docstrings
@@ -47,7 +46,6 @@ def load_tf_weights_in_gpt2(model, config, gpt2_checkpoint_path):
"""
try:
import re
import numpy as np
import tensorflow as tf
except ImportError:
logger.error(

View File

@@ -26,7 +26,6 @@ from io import open
import torch
import torch.nn as nn
from torch.nn import CrossEntropyLoss
from torch.nn.parameter import Parameter
from .configuration_openai import OpenAIGPTConfig
from .file_utils import add_start_docstrings

View File

@@ -25,7 +25,7 @@ import os
import torch
import torch.nn.functional as F
from torch import nn
from torch.nn import CrossEntropyLoss, MSELoss
from torch.nn import CrossEntropyLoss
from .configuration_t5 import T5Config
from .file_utils import DUMMY_INPUTS, DUMMY_MASK, add_start_docstrings

View File

@@ -29,7 +29,6 @@ from .configuration_auto import (
XLMConfig,
XLNetConfig,
)
from .file_utils import add_start_docstrings
from .modeling_tf_albert import (
TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_MAP,
TFAlbertForMaskedLM,

View File

@@ -24,7 +24,7 @@ import tensorflow as tf
from .configuration_ctrl import CTRLConfig
from .file_utils import add_start_docstrings
from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, get_initializer, shape_list
from .modeling_tf_utils import TFPreTrainedModel, TFSharedEmbeddings, shape_list
logger = logging.getLogger(__name__)

View File

@@ -16,7 +16,6 @@
"""
from __future__ import absolute_import, division, print_function, unicode_literals
import itertools
import logging
import math

View File

@@ -75,8 +75,8 @@ def load_pytorch_checkpoint_in_tf2_model(tf_model, pytorch_checkpoint_path, tf_i
""" Load pytorch checkpoints in a TF 2.0 model
"""
try:
import tensorflow as tf
import torch
import tensorflow as tf # noqa: F401
import torch # noqa: F401
except ImportError as e:
logger.error(
"Loading a PyTorch model in TensorFlow, requires both PyTorch and TensorFlow to be installed. Please see "
@@ -109,8 +109,8 @@ def load_pytorch_weights_in_tf2_model(tf_model, pt_state_dict, tf_inputs=None, a
""" Load pytorch state_dict in a TF 2.0 model.
"""
try:
import torch
import tensorflow as tf
import torch # noqa: F401
import tensorflow as tf # noqa: F401
from tensorflow.python.keras import backend as K
except ImportError as e:
logger.error(
@@ -208,8 +208,8 @@ def load_tf2_checkpoint_in_pytorch_model(pt_model, tf_checkpoint_path, tf_inputs
(see https://github.com/tensorflow/tensorflow/blob/ee16fcac960ae660e0e4496658a366e2f745e1f0/tensorflow/python/keras/engine/network.py#L1352-L1357).
"""
try:
import tensorflow as tf
import torch
import tensorflow as tf # noqa: F401
import torch # noqa: F401
except ImportError as e:
logger.error(
"Loading a TensorFlow model in PyTorch, requires both PyTorch and TensorFlow to be installed. Please see "
@@ -250,8 +250,8 @@ def load_tf2_weights_in_pytorch_model(pt_model, tf_weights, allow_missing_keys=F
""" Load TF2.0 symbolic weights in a PyTorch model
"""
try:
import tensorflow as tf
import torch
import tensorflow as tf # noqa: F401
import torch # noqa: F401
except ImportError as e:
logger.error(
"Loading a TensorFlow model in PyTorch, requires both PyTorch and TensorFlow to be installed. Please see "

View File

@@ -23,7 +23,7 @@ import tensorflow as tf
from .configuration_roberta import RobertaConfig
from .file_utils import add_start_docstrings
from .modeling_tf_bert import TFBertEmbeddings, TFBertMainLayer, gelu, gelu_new
from .modeling_tf_bert import TFBertEmbeddings, TFBertMainLayer, gelu
from .modeling_tf_utils import TFPreTrainedModel, get_initializer, shape_list

View File

@@ -20,13 +20,12 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import logging
import numpy as np
import tensorflow as tf
from .configuration_transfo_xl import TransfoXLConfig
from .file_utils import add_start_docstrings
from .modeling_tf_transfo_xl_utilities import TFAdaptiveSoftmaxMask
from .modeling_tf_utils import TFConv1D, TFPreTrainedModel, TFSequenceSummary, get_initializer, shape_list
from .modeling_tf_utils import TFPreTrainedModel, get_initializer, shape_list
logger = logging.getLogger(__name__)

View File

@@ -17,7 +17,6 @@
"""
import numpy as np
import tensorflow as tf
from .modeling_tf_utils import shape_list

View File

@@ -25,13 +25,11 @@ import logging
import torch
import torch.nn as nn
import torch.nn.functional as F
from torch.nn import CrossEntropyLoss
from torch.nn.parameter import Parameter
from .configuration_transfo_xl import TransfoXLConfig
from .file_utils import add_start_docstrings
from .modeling_transfo_xl_utilities import LogUniformSampler, ProjectedAdaptiveLogSoftmax, sample_logits
from .modeling_utils import Conv1D, PreTrainedModel, SequenceSummary, prune_conv1d_layer
from .modeling_utils import PreTrainedModel
logger = logging.getLogger(__name__)

View File

@@ -18,7 +18,6 @@
"""
import numpy as np
import torch
import torch.nn as nn
import torch.nn.functional as F

View File

@@ -20,7 +20,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import logging
import os
import six
import torch
from torch import nn
from torch.nn import CrossEntropyLoss

View File

@@ -22,7 +22,6 @@ import pickle
import sys
from abc import ABC, abstractmethod
from contextlib import contextmanager
from itertools import groupby
from os.path import abspath, exists
from typing import Dict, List, Optional, Tuple, Union

View File

@@ -37,9 +37,6 @@ if is_torch_available():
)
from transformers.modeling_bert import BERT_PRETRAINED_MODEL_ARCHIVE_MAP
from .modeling_common_test import CommonTestCases, ids_tensor
from .configuration_common_test import ConfigTester
@require_torch
class AutoModelTest(unittest.TestCase):

View File

@@ -20,7 +20,7 @@ from transformers import is_torch_available
from .configuration_common_test import ConfigTester
from .modeling_common_test import CommonTestCases, ids_tensor
from .utils import CACHE_DIR, require_torch, slow, torch_device
from .utils import require_torch, torch_device
if is_torch_available():

View File

@@ -19,8 +19,8 @@ import unittest
from transformers import is_torch_available
from .configuration_common_test import ConfigTester
from .modeling_common_test import CommonTestCases, floats_tensor, ids_tensor
from .utils import CACHE_DIR, require_torch, slow, torch_device
from .modeling_common_test import CommonTestCases, ids_tensor
from .utils import CACHE_DIR, require_torch, slow
if is_torch_available():

View File

@@ -24,7 +24,6 @@ from .utils import CACHE_DIR, require_tf, slow
if is_tf_available():
import tensorflow as tf
from transformers.modeling_tf_albert import (
TFAlbertModel,
TFAlbertForMaskedLM,

View File

@@ -35,10 +35,6 @@ if is_tf_available():
TFAutoModelForQuestionAnswering,
TFBertForQuestionAnswering,
)
from transformers.modeling_tf_bert import TF_BERT_PRETRAINED_MODEL_ARCHIVE_MAP
from .modeling_common_test import CommonTestCases, ids_tensor
from .configuration_common_test import ConfigTester
@require_tf

View File

@@ -24,13 +24,12 @@ import unittest
from transformers import is_tf_available, is_torch_available
from .utils import require_tf, slow
from .utils import require_tf
if is_tf_available():
import tensorflow as tf
import numpy as np
from transformers import TFPreTrainedModel
# from transformers.modeling_bert import BertModel, BertConfig, BERT_PRETRAINED_MODEL_ARCHIVE_MAP

View File

@@ -24,7 +24,6 @@ from .utils import CACHE_DIR, require_tf, slow
if is_tf_available():
import tensorflow as tf
from transformers.modeling_tf_ctrl import TFCTRLModel, TFCTRLLMHeadModel, TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP

View File

@@ -20,11 +20,10 @@ from transformers import DistilBertConfig, is_tf_available
from .configuration_common_test import ConfigTester
from .modeling_tf_common_test import TFCommonTestCases, ids_tensor
from .utils import CACHE_DIR, require_tf, slow
from .utils import require_tf
if is_tf_available():
import tensorflow as tf
from transformers.modeling_tf_distilbert import (
TFDistilBertModel,
TFDistilBertForMaskedLM,

View File

@@ -24,8 +24,7 @@ from .utils import CACHE_DIR, require_tf, slow
if is_tf_available():
import tensorflow as tf
from transformers.modeling_tf_t5 import TFT5Model, TFT5WithLMHeadModel, TF_T5_PRETRAINED_MODEL_ARCHIVE_MAP
from transformers.modeling_tf_t5 import TFT5Model, TFT5WithLMHeadModel
@require_tf

View File

@@ -17,7 +17,7 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import os
import unittest
from transformers.tokenization_albert import SPIECE_UNDERLINE, AlbertTokenizer
from transformers.tokenization_albert import AlbertTokenizer
from .tokenization_tests_commons import CommonTestCases

View File

@@ -19,7 +19,6 @@ import unittest
from transformers.tokenization_distilbert import DistilBertTokenizer
from .tokenization_bert_test import BertTokenizationTest
from .tokenization_tests_commons import CommonTestCases
from .utils import slow

View File

@@ -25,7 +25,6 @@ from .utils import require_torch
if is_torch_available():
import torch
from transformers.tokenization_transfo_xl import TransfoXLTokenizer, VOCAB_FILES_NAMES

View File

@@ -24,7 +24,6 @@ import unicodedata
import six
from .tokenization_bert import BasicTokenizer, BertTokenizer, WordpieceTokenizer, load_vocab
from .tokenization_utils import PreTrainedTokenizer
logger = logging.getLogger(__name__)

View File

@@ -17,7 +17,6 @@
from __future__ import absolute_import, division, print_function, unicode_literals
import logging
import unicodedata
from .tokenization_bert import BertTokenizer

View File

@@ -17,8 +17,6 @@ from __future__ import absolute_import, division, print_function, unicode_litera
import logging
import regex as re
from .tokenization_gpt2 import GPT2Tokenizer

View File

@@ -25,7 +25,6 @@ from io import open
import sacremoses as sm
from .tokenization_bert import BasicTokenizer
from .tokenization_utils import PreTrainedTokenizer