mirror of
https://github.com/immich-app/immich.git
synced 2025-07-09 09:12:57 +02:00
chore(ml): update pydantic (#13230)
* update pydantic * fix typing * remove unused import * remove unused schema
This commit is contained in:
parent
f29fb1655a
commit
e7397f35c9
6 changed files with 186 additions and 82 deletions
machine-learning/app
|
@ -810,11 +810,26 @@ class TestLoad:
|
|||
mock_model.model_format = ModelFormat.ONNX
|
||||
|
||||
|
||||
def test_root_endpoint(deployed_app: TestClient) -> None:
|
||||
response = deployed_app.get("http://localhost:3003")
|
||||
|
||||
body = response.json()
|
||||
assert response.status_code == 200
|
||||
assert body == {"message": "Immich ML"}
|
||||
|
||||
|
||||
def test_ping_endpoint(deployed_app: TestClient) -> None:
|
||||
response = deployed_app.get("http://localhost:3003/ping")
|
||||
|
||||
assert response.status_code == 200
|
||||
assert response.text == "pong"
|
||||
|
||||
|
||||
@pytest.mark.skipif(
|
||||
not settings.test_full,
|
||||
reason="More time-consuming since it deploys the app and loads models.",
|
||||
)
|
||||
class TestEndpoints:
|
||||
class TestPredictionEndpoints:
|
||||
def test_clip_image_endpoint(
|
||||
self, pil_image: Image.Image, responses: dict[str, Any], deployed_app: TestClient
|
||||
) -> None:
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue