diff --git a/great_expectations/experimental/metric_repository/cloud_data_store.py b/great_expectations/experimental/metric_repository/cloud_data_store.py index 2a107d36244a..1e50f12812eb 100644 --- a/great_expectations/experimental/metric_repository/cloud_data_store.py +++ b/great_expectations/experimental/metric_repository/cloud_data_store.py @@ -3,7 +3,7 @@ import uuid from typing import TYPE_CHECKING, Any, Dict, TypeVar -from great_expectations.compatibility.pydantic import BaseModel, Extra +from great_expectations.compatibility.pydantic import BaseModel from great_expectations.compatibility.typing_extensions import override from great_expectations.core.http import create_session from great_expectations.experimental.metric_repository.data_store import DataStore @@ -25,7 +25,7 @@ class PayloadData(BaseModel): attributes: Dict[str, Any] class Config: - extra = Extra.forbid + extra = "forbid" def orjson_dumps(v, *, default): @@ -49,7 +49,7 @@ class Payload(BaseModel): data: PayloadData class Config: - extra = Extra.forbid + extra = "forbid" json_dumps = orjson_dumps json_loads = orjson_loads diff --git a/great_expectations/experimental/metric_repository/metrics.py b/great_expectations/experimental/metric_repository/metrics.py index f039495dc399..c1b99b668114 100644 --- a/great_expectations/experimental/metric_repository/metrics.py +++ b/great_expectations/experimental/metric_repository/metrics.py @@ -15,7 +15,6 @@ Union, ) -from great_expectations.compatibility import pydantic from great_expectations.compatibility.pydantic import BaseModel, Field from great_expectations.compatibility.typing_extensions import override @@ -28,7 +27,7 @@ class MetricRepositoryBaseModel(BaseModel): """Base class for all MetricRepository related models.""" class Config: - extra = pydantic.Extra.forbid + extra = "forbid" class MetricException(MetricRepositoryBaseModel):