Updates pydantic config setting to use proper settings approach

DAGWorks-Inc · Dec 10, 2024 · bd72087 · bd72087
1 parent edcf1ab
commit bd72087
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/tests/integrations/test_burr_pydantic.py b/tests/integrations/test_burr_pydantic.py
@@ -3,7 +3,7 @@
 
 import pydantic
 import pytest
-from pydantic import BaseModel, EmailStr, Field
+from pydantic import BaseModel, ConfigDict, EmailStr, Field
 from pydantic.fields import FieldInfo
 
 from burr.core import expr
@@ -110,8 +110,7 @@ class MyModelWithConfig(pydantic.BaseModel):
         foo: int
         arbitrary: Arbitrary
 
-        class Config:
-            arbitrary_types_allowed = True
+        model_config = ConfigDict(arbitrary_types_allowed=True)
 
     SubsetModel = subset_model(MyModelWithConfig, ["foo", "bar"], [], "Subset")
     assert SubsetModel.__name__ == "MyModelWithConfigSubset"