From 5004d5af42c61c91d5df07aa139d37599ceb6215 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Mon, 27 Jan 2020 19:58:00 -0500 Subject: [PATCH] [serving] Update dependencies --- setup.py | 2 +- src/transformers/commands/serving.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index 4c908fe5e..6f1bf280f 100644 --- a/setup.py +++ b/setup.py @@ -63,7 +63,7 @@ extras["sklearn"] = ["scikit-learn"] extras["tf"] = ["tensorflow"] extras["torch"] = ["torch"] -extras["serving"] = ["pydantic", "uvicorn", "fastapi"] +extras["serving"] = ["pydantic", "uvicorn", "fastapi", "starlette"] extras["all"] = extras["serving"] + ["tensorflow", "torch"] extras["testing"] = ["pytest", "pytest-xdist"] diff --git a/src/transformers/commands/serving.py b/src/transformers/commands/serving.py index 6dbdece74..48d5ecb95 100644 --- a/src/transformers/commands/serving.py +++ b/src/transformers/commands/serving.py @@ -2,7 +2,6 @@ import logging 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 @@ -13,6 +12,7 @@ try: from fastapi import FastAPI, HTTPException, Body from fastapi.routing import APIRoute from pydantic import BaseModel + from starlette.responses import JSONResponse _serve_dependancies_installed = True except (ImportError, AttributeError):