From 647e448092cda9fb53073344ef5d8ac792ff1907 Mon Sep 17 00:00:00 2001 From: Andras Schmelczer Date: Sun, 3 Jul 2022 13:55:10 +0200 Subject: [PATCH] Remove trailing slash --- src/great_ai/great_ai/deploy/great_ai.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/great_ai/great_ai/deploy/great_ai.py b/src/great_ai/great_ai/deploy/great_ai.py index 0f306e6..8ef2835 100644 --- a/src/great_ai/great_ai/deploy/great_ai.py +++ b/src/great_ai/great_ai/deploy/great_ai.py @@ -183,13 +183,14 @@ class GreatAI(Generic[T]): def _bootstrap_prediction_endpoint(self) -> None: router = APIRouter( - prefix="/predict", tags=["predictions"], ) schema = self._get_schema() - @router.post("/", status_code=status.HTTP_200_OK, response_model=Trace[T]) + @router.post( + "/predict", status_code=status.HTTP_200_OK, response_model=Trace[T] + ) @use_http_exceptions def predict(input_value: schema) -> Trace[T]: # type: ignore return self(**cast(BaseModel, input_value).dict())