fix(ml): pydantic dep causes starting up issue (immich-app#11773)

* fix(ml): pydantic dep causes starting up issue * revert import
cxmlg · Aug 14, 2024 · fb962f4 · fb962f4
1 parent 7f7fec2
commit fb962f4
Show file tree

Hide file tree

Showing 5 changed files with 54 additions and 129 deletions.
diff --git a/machine-learning/app/config.py b/machine-learning/app/config.py
@@ -6,7 +6,7 @@
 from socket import socket
 
 from gunicorn.arbiter import Arbiter
-from pydantic.v1 import BaseModel, BaseSettings
+from pydantic import BaseModel, BaseSettings
 from rich.console import Console
 from rich.logging import RichHandler
 from uvicorn import Server

diff --git a/machine-learning/app/main.py b/machine-learning/app/main.py
@@ -15,7 +15,7 @@
 from fastapi.responses import ORJSONResponse
 from onnxruntime.capi.onnxruntime_pybind11_state import InvalidProtobuf, NoSuchFile
 from PIL.Image import Image
-from pydantic.v1 import ValidationError
+from pydantic import ValidationError
 from starlette.formparsers import MultiPartParser
 
 from app.models import get_model_deps

diff --git a/machine-learning/app/schemas.py b/machine-learning/app/schemas.py
@@ -3,7 +3,7 @@
 
 import numpy as np
 import numpy.typing as npt
-from pydantic.v1 import BaseModel
+from pydantic import BaseModel
 
 
 class StrEnum(str, Enum):