aboutsummaryrefslogtreecommitdiff
path: root/tests/incite/collections/test_df_collection_thl_marketplaces.py
diff options
context:
space:
mode:
Diffstat (limited to 'tests/incite/collections/test_df_collection_thl_marketplaces.py')
-rw-r--r--tests/incite/collections/test_df_collection_thl_marketplaces.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/tests/incite/collections/test_df_collection_thl_marketplaces.py b/tests/incite/collections/test_df_collection_thl_marketplaces.py
index 0a77938..54c27f7 100644
--- a/tests/incite/collections/test_df_collection_thl_marketplaces.py
+++ b/tests/incite/collections/test_df_collection_thl_marketplaces.py
@@ -1,11 +1,11 @@
from datetime import datetime, timezone
from itertools import product
+from typing import TYPE_CHECKING
import pytest
-from pandera import Column, Index, DataFrameSchema
+from pandera import Column, DataFrameSchema, Index
-from generalresearch.incite.collections import DFCollection
-from generalresearch.incite.collections import DFCollectionType
+from generalresearch.incite.collections import DFCollection, DFCollectionType
from generalresearch.incite.collections.thl_marketplaces import (
InnovateSurveyHistoryCollection,
MorningSurveyTimeseriesCollection,
@@ -14,6 +14,9 @@ from generalresearch.incite.collections.thl_marketplaces import (
)
from test_utils.incite.conftest import mnt_filepath
+if TYPE_CHECKING:
+ from generalresearch.incite.base import GRLDatasets
+
def combo_object():
for x in product(