diff --git a/pyproject.toml b/pyproject.toml index 01501c6..19179d0 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -26,7 +26,7 @@ dependencies = [ "numpy", "scipy", "matplotlib", - "pydantic", + "pydantic>=2.0.0", ] [project.urls] diff --git a/src/quera_ahs_utils/quera_ir/capabilities.py b/src/quera_ahs_utils/quera_ir/capabilities.py index e86a55f..1e1ca26 100644 --- a/src/quera_ahs_utils/quera_ir/capabilities.py +++ b/src/quera_ahs_utils/quera_ir/capabilities.py @@ -1,4 +1,4 @@ -from pydantic import BaseModel +from pydantic.v1 import BaseModel __all__ = ["QuEraCapabilities"] diff --git a/src/quera_ahs_utils/quera_ir/task_results.py b/src/quera_ahs_utils/quera_ir/task_results.py index be51f4c..3482a4b 100644 --- a/src/quera_ahs_utils/quera_ir/task_results.py +++ b/src/quera_ahs_utils/quera_ir/task_results.py @@ -1,5 +1,5 @@ from enum import Enum -from pydantic import BaseModel, conlist, conint +from pydantic.v1 import BaseModel, conlist, conint from typing import Callable, Optional, List, Tuple import numpy as np diff --git a/src/quera_ahs_utils/quera_ir/task_specification.py b/src/quera_ahs_utils/quera_ir/task_specification.py index 9f24d97..527b470 100644 --- a/src/quera_ahs_utils/quera_ir/task_specification.py +++ b/src/quera_ahs_utils/quera_ir/task_specification.py @@ -1,4 +1,4 @@ -from pydantic import BaseModel +from pydantic.v1 import BaseModel from typing import Optional, List, Tuple, Union from decimal import Decimal