fix: prepare pydantic v2

This commit is contained in:
Dowon
2024-03-16 22:52:32 +09:00
parent 2255341fb8
commit c3e2884b80
2 changed files with 23 additions and 11 deletions

View File

@@ -5,16 +5,28 @@ from dataclasses import dataclass
from functools import cached_property, partial
from typing import Any, Literal, NamedTuple, Optional
from pydantic import (
BaseModel,
Extra,
NonNegativeFloat,
NonNegativeInt,
PositiveInt,
confloat,
conint,
validator,
)
try:
from pydantic.v1 import (
BaseModel,
Extra,
NonNegativeFloat,
NonNegativeInt,
PositiveInt,
confloat,
conint,
validator,
)
except ImportError:
from pydantic import (
BaseModel,
Extra,
NonNegativeFloat,
NonNegativeInt,
PositiveInt,
confloat,
conint,
validator,
)
@dataclass

View File

@@ -8,7 +8,7 @@ license = { text = "AGPL-3.0" }
dependencies = [
"ultralytics>=8.1",
"mediapipe>=10",
"pydantic<2",
"pydantic<3",
"rich>=13",
"huggingface_hub",
]