Skip to content

Commit

Permalink
Deprecation of SchemaModel in Pandera
Browse files Browse the repository at this point in the history
Signed-off-by: miguelgfierro <[email protected]>
  • Loading branch information
miguelgfierro committed Jun 29, 2024
1 parent fe13790 commit cf64eed
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion recommenders/datasets/movielens.py
Original file line number Diff line number Diff line change
Expand Up @@ -582,7 +582,7 @@ def unique_columns(df, *, columns):
return not df[columns].duplicated().any()


class MockMovielensSchema(pa.SchemaModel):
class MockMovielensSchema(pa.DataFrameModel):
"""
Mock dataset schema to generate fake data for testing purpose.
This schema is configured to mimic the Movielens dataset
Expand Down

0 comments on commit cf64eed

Please sign in to comment.