From b3728fe6632ae9393dc2d1e7859b06f5538dc0a7 Mon Sep 17 00:00:00 2001 From: Matthew Grossman Date: Thu, 8 Aug 2024 12:58:34 -0700 Subject: [PATCH] PLAT-2332: Upgrade to pydantic v2 Co-authored-by: Gretel GitHub Bot GitOrigin-RevId: 402497789adbd9d2bbf30f8de11c736c43a7993b --- requirements.txt | 2 +- src/gretel_trainer/strategy.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/requirements.txt b/requirements.txt index 7731cb78..0e03868a 100644 --- 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 index c9949966..85e21be1 100644 --- 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):