PLAT-2332: Upgrade to pydantic v2

Co-authored-by: Gretel GitHub Bot <[email protected]> GitOrigin-RevId: 402497789adbd9d2bbf30f8de11c736c43a7993b
gretelai · Aug 8, 2024 · b3728fe · b3728fe
1 parent d68f562
commit b3728fe
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,7 +1,7 @@
 boto3~=1.20
 gretel-client>=0.17.7
 pandas~=1.5
-pydantic~=1.9
+pydantic>=2
 requests~=2.25
 smart_open[s3]~=5.2
 typing_extensions~=4.7
diff --git a/src/gretel_trainer/strategy.py b/src/gretel_trainer/strategy.py
@@ -9,7 +9,7 @@
 
 import pandas as pd
 
-from pydantic import BaseModel, Field, PrivateAttr
+from pydantic.v1 import BaseModel, Field, PrivateAttr
 
 
 class RowPartition(BaseModel):