From 403f0e0629618e3aa51d55352c1b4135940f6fab Mon Sep 17 00:00:00 2001 From: Raphael Glon Date: Mon, 30 Sep 2024 09:44:20 +0200 Subject: [PATCH 1/2] Reclassify model loading error as server errors Signed-off-by: Raphael Glon --- api_inference_community/routes.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/api_inference_community/routes.py b/api_inference_community/routes.py index 3fa20e66..347cf8ea 100644 --- a/api_inference_community/routes.py +++ b/api_inference_community/routes.py @@ -119,8 +119,6 @@ async def pipeline_route(request: Request) -> Response: f'{error["msg"]}: received `{error["input"]}` in `parameters`' ) return JSONResponse({"error": errors}, status_code=400) - except (EnvironmentError, ValueError) as e: - return JSONResponse({"error": str(e)}, status_code=400) except Exception as e: return JSONResponse({"error": str(e)}, status_code=500) From 5a3785473de2fe102ddd85b93b317dbf734ad2e6 Mon Sep 17 00:00:00 2001 From: Raphael Glon Date: Mon, 30 Sep 2024 10:02:35 +0200 Subject: [PATCH 2/2] adapt unittests Signed-off-by: Raphael Glon --- tests/test_routes.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_routes.py b/tests/test_routes.py index a4516079..05c9cae7 100644 --- a/tests/test_routes.py +++ b/tests/test_routes.py @@ -99,7 +99,7 @@ async def startup_event(): self.assertEqual( response.status_code, - 400, + 500, ) self.assertEqual( response.content,