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

* fix(ml): pydantic dep causes starting up issue * revert import
2024-08-14 10:20:12 -05:00
parent 7f7fec2cea
commit fb962f49ea
5 changed files with 54 additions and 129 deletions
@@ -3,7 +3,7 @@ from typing import Any, Literal, Protocol, TypedDict, TypeGuard, TypeVar

 import numpy as np
 import numpy.typing as npt
-from pydantic.v1 import BaseModel
+from pydantic import BaseModel


 class StrEnum(str, Enum):