Skip to content

Commit 45907cd

Browse files
authored
Reclassify model loading error as server errors (#457)
* Reclassify model loading error as server errors * adapt unittests
1 parent 8b71cc4 commit 45907cd

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

api_inference_community/routes.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -119,8 +119,6 @@ async def pipeline_route(request: Request) -> Response:
119119
f'{error["msg"]}: received `{error["input"]}` in `parameters`'
120120
)
121121
return JSONResponse({"error": errors}, status_code=400)
122-
except (EnvironmentError, ValueError) as e:
123-
return JSONResponse({"error": str(e)}, status_code=400)
124122
except Exception as e:
125123
return JSONResponse({"error": str(e)}, status_code=500)
126124

tests/test_routes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -99,7 +99,7 @@ async def startup_event():
9999

100100
self.assertEqual(
101101
response.status_code,
102-
400,
102+
500,
103103
)
104104
self.assertEqual(
105105
response.content,

0 commit comments

Comments
 (0)