style and quality

6b081f04 · thomwolf · Lysandre Debut · 0e31e06a · 6b081f04 · 6b081f04
Commit 6b081f04 authored Jan 24, 2020 by thomwolf Committed by Lysandre Debut Jan 27, 2020
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

src/transformers/modeling_auto.py src/transformers/modeling_auto.py +1 -1

src/transformers/modeling_tf_auto.py src/transformers/modeling_tf_auto.py +1 -1

No files found.
--- a/src/transformers/modeling_auto.py
+++ b/src/transformers/modeling_auto.py
@@ -45,11 +45,11 @@ from .modeling_albert import (
 from .modeling_bert import (
    BERT_PRETRAINED_MODEL_ARCHIVE_MAP,
    BertForMaskedLM,
+    BertForPreTraining,
    BertForQuestionAnswering,
    BertForSequenceClassification,
    BertForTokenClassification,
    BertModel,
-    BertForPreTraining,
 )
 from .modeling_camembert import (
    CAMEMBERT_PRETRAINED_MODEL_ARCHIVE_MAP,

--- a/src/transformers/modeling_tf_auto.py
+++ b/src/transformers/modeling_tf_auto.py
@@ -42,11 +42,11 @@ from .modeling_tf_albert import (
 from .modeling_tf_bert import (
    TF_BERT_PRETRAINED_MODEL_ARCHIVE_MAP,
    TFBertForMaskedLM,
+    TFBertForPreTraining,
    TFBertForQuestionAnswering,
    TFBertForSequenceClassification,
    TFBertForTokenClassification,
    TFBertModel,
-    TFBertForPreTraining,
 )
 from .modeling_tf_ctrl import TF_CTRL_PRETRAINED_MODEL_ARCHIVE_MAP, TFCTRLLMHeadModel, TFCTRLModel
 from .modeling_tf_distilbert import (