revert import

immich-app · Aug 14, 2024 · 4809aa4 · 4809aa4
1 parent 6f5ed55
commit 4809aa4
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 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):