Fix F401 flake8 warning (x28).

Do manually what autoflake couldn't manage.
2019-12-21 22:04:51 +01:00
parent 783a616999
commit 939148b050
11 changed files with 4 additions and 56 deletions
--- a/templates/adding_a_new_model/tests/modeling_tf_xxx_test.py
+++ b/templates/adding_a_new_model/tests/modeling_tf_xxx_test.py
@@ -30,7 +30,6 @@ if is_tf_available():
        TFXxxForSequenceClassification,
        TFXxxForTokenClassification,
        TFXxxForQuestionAnswering,
-        TF_XXX_PRETRAINED_MODEL_ARCHIVE_MAP,
    )


--- a/templates/adding_a_new_model/tests/modeling_xxx_test.py
+++ b/templates/adding_a_new_model/tests/modeling_xxx_test.py
@@ -28,12 +28,9 @@ if is_torch_available():
        XxxConfig,
        XxxModel,
        XxxForMaskedLM,
-        XxxForNextSentencePrediction,
-        XxxForPreTraining,
        XxxForQuestionAnswering,
        XxxForSequenceClassification,
        XxxForTokenClassification,
-        XxxForMultipleChoice,
    )
    from transformers.modeling_xxx import XXX_PRETRAINED_MODEL_ARCHIVE_MAP