diff options
Diffstat (limited to 'tests/incite')
5 files changed, 5 insertions, 5 deletions
diff --git a/tests/incite/collections/test_df_collection_base.py b/tests/incite/collections/test_df_collection_base.py index 10f025b..31d1720 100644 --- a/tests/incite/collections/test_df_collection_base.py +++ b/tests/incite/collections/test_df_collection_base.py @@ -3,7 +3,7 @@ from typing import TYPE_CHECKING import pandas as pd import pytest -from pandera import DataFrameSchema +from pandera.pandas import DataFrameSchema from generalresearch.incite.collections import ( DFCollection, diff --git a/tests/incite/collections/test_df_collection_item_thl_web.py b/tests/incite/collections/test_df_collection_item_thl_web.py index 29f3677..a858fbe 100644 --- a/tests/incite/collections/test_df_collection_item_thl_web.py +++ b/tests/incite/collections/test_df_collection_item_thl_web.py @@ -21,7 +21,7 @@ from distributed.utils_test import ( loop_in_thread, ) from faker import Faker -from pandera import DataFrameSchema +from pandera.pandas import DataFrameSchema from pydantic import FilePath from generalresearch.incite.base import CollectionItemBase diff --git a/tests/incite/collections/test_df_collection_thl_marketplaces.py b/tests/incite/collections/test_df_collection_thl_marketplaces.py index 54c27f7..8ce8acc 100644 --- a/tests/incite/collections/test_df_collection_thl_marketplaces.py +++ b/tests/incite/collections/test_df_collection_thl_marketplaces.py @@ -3,7 +3,7 @@ from itertools import product from typing import TYPE_CHECKING import pytest -from pandera import Column, DataFrameSchema, Index +from pandera.pandas import Column, DataFrameSchema, Index from generalresearch.incite.collections import DFCollection, DFCollectionType from generalresearch.incite.collections.thl_marketplaces import ( diff --git a/tests/incite/collections/test_df_collection_thl_web.py b/tests/incite/collections/test_df_collection_thl_web.py index 51ca128..c64dac8 100644 --- a/tests/incite/collections/test_df_collection_thl_web.py +++ b/tests/incite/collections/test_df_collection_thl_web.py @@ -5,7 +5,7 @@ from typing import TYPE_CHECKING import dask.dataframe as dd import pandas as pd import pytest -from pandera import DataFrameSchema +from pandera.pandas import DataFrameSchema from generalresearch.incite.collections import DFCollection, DFCollectionType diff --git a/tests/incite/mergers/test_merge_collection.py b/tests/incite/mergers/test_merge_collection.py index 692cac3..ec507bc 100644 --- a/tests/incite/mergers/test_merge_collection.py +++ b/tests/incite/mergers/test_merge_collection.py @@ -3,7 +3,7 @@ from itertools import product import pandas as pd import pytest -from pandera import DataFrameSchema +from pandera.pandas import DataFrameSchema from generalresearch.incite.mergers import ( MergeCollection, |