From 4809aa43bbece6bb25e083c15cec1cd72d474984 Mon Sep 17 00:00:00 2001 From: Alex Tran Date: Wed, 14 Aug 2024 10:01:34 -0500 Subject: [PATCH] revert import --- machine-learning/app/config.py | 2 +- machine-learning/app/main.py | 2 +- machine-learning/app/schemas.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/machine-learning/app/config.py b/machine-learning/app/config.py index 5dec031529826..af2d0aa4b91a9 100644 --- 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 index 52b9a66c052e8..000119937e74a 100644 --- 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 index e8a36ef44dcf3..f051db12c3d4d 100644 --- 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):