Unverified Commit a5737779 authored by Sylvain Gugger's avatar Sylvain Gugger Committed by GitHub
Browse files

Update repo to isort v5 (#6686)

* Run new isort

* More changes

* Update CI, CONTRIBUTING and benchmarks
parent d329c9b0
...@@ -25,18 +25,22 @@ from .test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor, r ...@@ -25,18 +25,22 @@ from .test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor, r
if is_torch_available(): if is_torch_available():
import torch import torch
from transformers import ( from transformers import (
RobertaConfig, RobertaConfig,
RobertaModel,
RobertaForCausalLM, RobertaForCausalLM,
RobertaForMaskedLM, RobertaForMaskedLM,
RobertaForMultipleChoice, RobertaForMultipleChoice,
RobertaForQuestionAnswering, RobertaForQuestionAnswering,
RobertaForSequenceClassification, RobertaForSequenceClassification,
RobertaForTokenClassification, RobertaForTokenClassification,
RobertaModel,
)
from transformers.modeling_roberta import (
ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
RobertaEmbeddings,
create_position_ids_from_input_ids,
) )
from transformers.modeling_roberta import RobertaEmbeddings, create_position_ids_from_input_ids
from transformers.modeling_roberta import ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST
class RobertaModelTester: class RobertaModelTester:
......
...@@ -28,7 +28,8 @@ from .test_modeling_common import ModelTesterMixin, ids_tensor ...@@ -28,7 +28,8 @@ from .test_modeling_common import ModelTesterMixin, ids_tensor
if is_torch_available(): if is_torch_available():
import torch import torch
from transformers import T5Config, T5Model, T5ForConditionalGeneration
from transformers import T5Config, T5ForConditionalGeneration, T5Model
from transformers.modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_LIST from transformers.modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.tokenization_t5 import T5Tokenizer from transformers.tokenization_t5 import T5Tokenizer
......
...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_albert import ( from transformers.modeling_tf_albert import (
TFAlbertModel, TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFAlbertForPreTraining,
TFAlbertForMaskedLM, TFAlbertForMaskedLM,
TFAlbertForMultipleChoice, TFAlbertForMultipleChoice,
TFAlbertForSequenceClassification, TFAlbertForPreTraining,
TFAlbertForQuestionAnswering, TFAlbertForQuestionAnswering,
TFAlbertForSequenceClassification,
TFAlbertForTokenClassification, TFAlbertForTokenClassification,
TF_ALBERT_PRETRAINED_MODEL_ARCHIVE_LIST, TFAlbertModel,
) )
......
...@@ -27,36 +27,36 @@ if is_tf_available(): ...@@ -27,36 +27,36 @@ if is_tf_available():
GPT2Config, GPT2Config,
T5Config, T5Config,
TFAutoModel, TFAutoModel,
TFBertModel, TFAutoModelForCausalLM,
TFAutoModelForMaskedLM,
TFAutoModelForPreTraining, TFAutoModelForPreTraining,
TFBertForPreTraining, TFAutoModelForQuestionAnswering,
TFAutoModelForSeq2SeqLM,
TFAutoModelForSequenceClassification,
TFAutoModelWithLMHead, TFAutoModelWithLMHead,
TFBertForMaskedLM, TFBertForMaskedLM,
TFRobertaForMaskedLM, TFBertForPreTraining,
TFAutoModelForSequenceClassification,
TFBertForSequenceClassification,
TFAutoModelForQuestionAnswering,
TFBertForQuestionAnswering, TFBertForQuestionAnswering,
TFAutoModelForCausalLM, TFBertForSequenceClassification,
TFBertModel,
TFGPT2LMHeadModel, TFGPT2LMHeadModel,
TFAutoModelForMaskedLM, TFRobertaForMaskedLM,
TFAutoModelForSeq2SeqLM,
TFT5ForConditionalGeneration, TFT5ForConditionalGeneration,
) )
from transformers.modeling_tf_bert import TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_gpt2 import TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_t5 import TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_auto import ( from transformers.modeling_tf_auto import (
TF_MODEL_MAPPING, TF_MODEL_FOR_CAUSAL_LM_MAPPING,
TF_MODEL_FOR_MASKED_LM_MAPPING,
TF_MODEL_FOR_PRETRAINING_MAPPING, TF_MODEL_FOR_PRETRAINING_MAPPING,
TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING, TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING,
TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING,
TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING, TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING,
TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING, TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING,
TF_MODEL_MAPPING,
TF_MODEL_WITH_LM_HEAD_MAPPING, TF_MODEL_WITH_LM_HEAD_MAPPING,
TF_MODEL_FOR_CAUSAL_LM_MAPPING,
TF_MODEL_FOR_MASKED_LM_MAPPING,
TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING,
) )
from transformers.modeling_tf_bert import TF_BERT_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_gpt2 import TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_t5 import TF_T5_PRETRAINED_MODEL_ARCHIVE_LIST
@require_tf @require_tf
......
...@@ -25,16 +25,17 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,16 +25,17 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_bert import ( from transformers.modeling_tf_bert import (
TFBertModel,
TFBertLMHeadModel,
TFBertForMaskedLM, TFBertForMaskedLM,
TFBertForMultipleChoice,
TFBertForNextSentencePrediction, TFBertForNextSentencePrediction,
TFBertForPreTraining, TFBertForPreTraining,
TFBertForQuestionAnswering,
TFBertForSequenceClassification, TFBertForSequenceClassification,
TFBertForMultipleChoice,
TFBertForTokenClassification, TFBertForTokenClassification,
TFBertForQuestionAnswering, TFBertLMHeadModel,
TFBertModel,
) )
......
...@@ -20,8 +20,9 @@ from transformers.testing_utils import require_tf, slow ...@@ -20,8 +20,9 @@ from transformers.testing_utils import require_tf, slow
if is_tf_available(): if is_tf_available():
import tensorflow as tf
import numpy as np import numpy as np
import tensorflow as tf
from transformers import TFCamembertModel from transformers import TFCamembertModel
......
...@@ -28,20 +28,20 @@ from transformers.testing_utils import _tf_gpu_memory_limit, require_tf, slow ...@@ -28,20 +28,20 @@ from transformers.testing_utils import _tf_gpu_memory_limit, require_tf, slow
if is_tf_available(): if is_tf_available():
import tensorflow as tf
import numpy as np import numpy as np
import tensorflow as tf
from transformers import ( from transformers import (
tf_top_k_top_p_filtering, TF_MODEL_FOR_CAUSAL_LM_MAPPING,
TFAdaptiveEmbedding, TF_MODEL_FOR_MASKED_LM_MAPPING,
TFSharedEmbeddings,
TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING, TF_MODEL_FOR_MULTIPLE_CHOICE_MAPPING,
TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING, TF_MODEL_FOR_QUESTION_ANSWERING_MAPPING,
TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING,
TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING, TF_MODEL_FOR_SEQUENCE_CLASSIFICATION_MAPPING,
TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING, TF_MODEL_FOR_TOKEN_CLASSIFICATION_MAPPING,
TF_MODEL_FOR_CAUSAL_LM_MAPPING, TFAdaptiveEmbedding,
TF_MODEL_FOR_MASKED_LM_MAPPING, TFSharedEmbeddings,
TF_MODEL_FOR_SEQ_TO_SEQ_CAUSAL_LM_MAPPING, tf_top_k_top_p_filtering,
) )
if _tf_gpu_memory_limit is not None: if _tf_gpu_memory_limit is not None:
...@@ -260,6 +260,7 @@ class TFModelTesterMixin: ...@@ -260,6 +260,7 @@ class TFModelTesterMixin:
return return
import torch import torch
import transformers import transformers
config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common() config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()
......
...@@ -25,7 +25,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,7 +25,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_ctrl import TFCTRLModel, TFCTRLLMHeadModel, TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST
from transformers.modeling_tf_ctrl import TF_CTRL_PRETRAINED_MODEL_ARCHIVE_LIST, TFCTRLLMHeadModel, TFCTRLModel
class TFCTRLModelTester(object): class TFCTRLModelTester(object):
......
...@@ -25,14 +25,15 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,14 +25,15 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_distilbert import ( from transformers.modeling_tf_distilbert import (
TFDistilBertModel, TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFDistilBertForMaskedLM, TFDistilBertForMaskedLM,
TFDistilBertForMultipleChoice,
TFDistilBertForQuestionAnswering, TFDistilBertForQuestionAnswering,
TFDistilBertForSequenceClassification, TFDistilBertForSequenceClassification,
TFDistilBertForTokenClassification, TFDistilBertForTokenClassification,
TFDistilBertForMultipleChoice, TFDistilBertModel,
TF_DISTILBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
) )
......
...@@ -27,13 +27,13 @@ if is_tf_available(): ...@@ -27,13 +27,13 @@ if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_electra import ( from transformers.modeling_tf_electra import (
TFElectraModel,
TFElectraForMaskedLM, TFElectraForMaskedLM,
TFElectraForMultipleChoice, TFElectraForMultipleChoice,
TFElectraForPreTraining, TFElectraForPreTraining,
TFElectraForQuestionAnswering,
TFElectraForSequenceClassification, TFElectraForSequenceClassification,
TFElectraForTokenClassification, TFElectraForTokenClassification,
TFElectraForQuestionAnswering, TFElectraModel,
) )
......
...@@ -23,18 +23,18 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -23,18 +23,18 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf
import numpy as np import numpy as np
import tensorflow as tf
from transformers import ( from transformers import (
TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
FlaubertConfig, FlaubertConfig,
TFFlaubertModel, TFFlaubertForMultipleChoice,
TFFlaubertWithLMHeadModel,
TFFlaubertForSequenceClassification,
TFFlaubertForQuestionAnsweringSimple, TFFlaubertForQuestionAnsweringSimple,
TFFlaubertForSequenceClassification,
TFFlaubertForTokenClassification, TFFlaubertForTokenClassification,
TFFlaubertForMultipleChoice, TFFlaubertModel,
TF_FLAUBERT_PRETRAINED_MODEL_ARCHIVE_LIST, TFFlaubertWithLMHeadModel,
) )
......
...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_gpt2 import ( from transformers.modeling_tf_gpt2 import (
TFGPT2Model,
TFGPT2LMHeadModel,
TFGPT2DoubleHeadsModel,
TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST, TF_GPT2_PRETRAINED_MODEL_ARCHIVE_LIST,
TFGPT2DoubleHeadsModel,
TFGPT2LMHeadModel,
TFGPT2Model,
shape_list, shape_list,
) )
......
...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers import ( from transformers import (
LongformerConfig, LongformerConfig,
TFLongformerModel,
TFLongformerForMaskedLM, TFLongformerForMaskedLM,
TFLongformerForQuestionAnswering, TFLongformerForQuestionAnswering,
TFLongformerModel,
TFLongformerSelfAttention, TFLongformerSelfAttention,
) )
......
...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_mobilebert import ( from transformers.modeling_tf_mobilebert import (
TFMobileBertModel,
TFMobileBertForMaskedLM, TFMobileBertForMaskedLM,
TFMobileBertForMultipleChoice,
TFMobileBertForNextSentencePrediction, TFMobileBertForNextSentencePrediction,
TFMobileBertForPreTraining, TFMobileBertForPreTraining,
TFMobileBertForQuestionAnswering,
TFMobileBertForSequenceClassification, TFMobileBertForSequenceClassification,
TFMobileBertForMultipleChoice,
TFMobileBertForTokenClassification, TFMobileBertForTokenClassification,
TFMobileBertForQuestionAnswering, TFMobileBertModel,
) )
......
...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,11 +25,12 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers.modeling_tf_openai import ( from transformers.modeling_tf_openai import (
TFOpenAIGPTModel,
TFOpenAIGPTLMHeadModel,
TFOpenAIGPTDoubleHeadsModel,
TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST, TF_OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_LIST,
TFOpenAIGPTDoubleHeadsModel,
TFOpenAIGPTLMHeadModel,
TFOpenAIGPTModel,
) )
......
...@@ -24,16 +24,17 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -24,16 +24,17 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf
import numpy import numpy
import tensorflow as tf
from transformers.modeling_tf_roberta import ( from transformers.modeling_tf_roberta import (
TFRobertaModel, TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
TFRobertaForMaskedLM, TFRobertaForMaskedLM,
TFRobertaForMultipleChoice,
TFRobertaForQuestionAnswering,
TFRobertaForSequenceClassification, TFRobertaForSequenceClassification,
TFRobertaForTokenClassification, TFRobertaForTokenClassification,
TFRobertaForQuestionAnswering, TFRobertaModel,
TFRobertaForMultipleChoice,
TF_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
) )
......
...@@ -26,7 +26,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -26,7 +26,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers import TFT5Model, TFT5ForConditionalGeneration, T5Tokenizer
from transformers import T5Tokenizer, TFT5ForConditionalGeneration, TFT5Model
class TFT5ModelTester: class TFT5ModelTester:
......
...@@ -26,11 +26,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -26,11 +26,8 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers import (
TFTransfoXLModel, from transformers import TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST, TFTransfoXLLMHeadModel, TFTransfoXLModel
TFTransfoXLLMHeadModel,
TF_TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_LIST,
)
class TFTransfoXLModelTester: class TFTransfoXLModelTester:
......
...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor ...@@ -25,15 +25,16 @@ from .test_modeling_tf_common import TFModelTesterMixin, ids_tensor
if is_tf_available(): if is_tf_available():
import tensorflow as tf import tensorflow as tf
from transformers import ( from transformers import (
XLMConfig, TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST,
TFXLMModel, TFXLMForMultipleChoice,
TFXLMWithLMHeadModel,
TFXLMForSequenceClassification,
TFXLMForQuestionAnsweringSimple, TFXLMForQuestionAnsweringSimple,
TFXLMForSequenceClassification,
TFXLMForTokenClassification, TFXLMForTokenClassification,
TFXLMForMultipleChoice, TFXLMModel,
TF_XLM_PRETRAINED_MODEL_ARCHIVE_LIST, TFXLMWithLMHeadModel,
XLMConfig,
) )
......
...@@ -20,8 +20,9 @@ from transformers.testing_utils import require_tf, slow ...@@ -20,8 +20,9 @@ from transformers.testing_utils import require_tf, slow
if is_tf_available(): if is_tf_available():
import tensorflow as tf
import numpy as np import numpy as np
import tensorflow as tf
from transformers import TFXLMRobertaModel from transformers import TFXLMRobertaModel
......
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