[MAINTENANCE] remove deprecated usage of pydantic Extra

great-expectations · Oct 30, 2023 · f255234 · f255234
1 parent ebd0519
commit f255234
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git 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
@@ -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):