Use pandera.DataFrameModel (#1703)

Signed-off-by: Eduardo Apolinario <[email protected]> Co-authored-by: Eduardo Apolinario <[email protected]>
flyteorg · Jul 4, 2024 · be6659d · be6659d
1 parent 5aee3d0
commit be6659d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py b/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py
@@ -118,7 +118,7 @@
 
 
 # %%
-class RawData(pa.SchemaModel):
+class RawData(pa.DataFrameModel):
     age: Series[int] = pa.Field(in_range={"min_value": 0, "max_value": 200})
     sex: Series[int] = pa.Field(isin=[0, 1])
     cp: Series[int] = pa.Field(

diff --git a/examples/pandera_plugin/requirements.in b/examples/pandera_plugin/requirements.in
@@ -6,5 +6,5 @@ flytekitplugins-pandera>=0.16.0
 hypothesis
 joblib
 pandas
-pandera>=0.7.0
+pandera>=0.20.0
 scikit-learn