diff --git a/setup.py b/setup.py index c8b39b652a..47550d6aa9 100644 --- a/setup.py +++ b/setup.py @@ -59,7 +59,7 @@ "scipy>=1.5.4", "requests>=2.21.0", "PyYAML>=5.1", - "pydantic<2", + "pydantic<3", "fastapi>=0.100.0", "fastapi-restful>=0.5.0", "typing-inspect>=0.9.0", diff --git a/tests/utils/test_pydantic_utils.py b/tests/utils/test_pydantic_utils.py index fd67130030..3907936ad1 100644 --- a/tests/utils/test_pydantic_utils.py +++ b/tests/utils/test_pydantic_utils.py @@ -1,8 +1,8 @@ from typing import Dict import pytest -from pydantic import parse_obj_as +from evidently._pydantic_compat import parse_obj_as from evidently.base_metric import MetricResult from evidently.pydantic_utils import PolymorphicModel