Reclassify model loading error as server errors (#457)

* Reclassify model loading error as server errors * adapt unittests
huggingface · Oct 1, 2024 · 45907cd · 45907cd
1 parent 8b71cc4
commit 45907cd
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git 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)
 

diff --git 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,