From ea56d305be0000b3f54ea378df77f92867779ba5 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Mon, 27 Jan 2020 12:13:32 -0500 Subject: [PATCH] make style --- src/transformers/commands/serving.py | 1 + src/transformers/modeling_auto.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/src/transformers/commands/serving.py b/src/transformers/commands/serving.py index 6dbdece74c..0d92f818c5 100644 --- a/src/transformers/commands/serving.py +++ b/src/transformers/commands/serving.py @@ -3,6 +3,7 @@ from argparse import ArgumentParser, Namespace from typing import Any, List, Optional from starlette.responses import JSONResponse + from transformers import Pipeline from transformers.commands import BaseTransformersCLICommand from transformers.pipelines import SUPPORTED_TASKS, pipeline diff --git a/src/transformers/modeling_auto.py b/src/transformers/modeling_auto.py index 1f8015f258..35aaa6e61a 100644 --- a/src/transformers/modeling_auto.py +++ b/src/transformers/modeling_auto.py @@ -71,10 +71,10 @@ from .modeling_openai import OPENAI_GPT_PRETRAINED_MODEL_ARCHIVE_MAP, OpenAIGPTL from .modeling_roberta import ( ROBERTA_PRETRAINED_MODEL_ARCHIVE_MAP, RobertaForMaskedLM, + RobertaForQuestionAnswering, RobertaForSequenceClassification, RobertaForTokenClassification, RobertaModel, - RobertaForQuestionAnswering ) from .modeling_t5 import T5_PRETRAINED_MODEL_ARCHIVE_MAP, T5Model, T5WithLMHeadModel from .modeling_transfo_xl import TRANSFO_XL_PRETRAINED_MODEL_ARCHIVE_MAP, TransfoXLLMHeadModel, TransfoXLModel