fix mypy

unionai-oss · Apr 29, 2022 · 9c5eaa3 · 9c5eaa3
1 parent f302b5a
commit 9c5eaa3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pandera/typing/__init__.py b/pandera/typing/__init__.py
@@ -4,6 +4,8 @@
 the typing module.
 """
 
+from typing import Set, Type
+
 from . import dask, fastapi, geopandas, modin, pyspark
 from .common import (
     BOOL,
@@ -39,9 +41,9 @@
 )
 from .pandas import DataFrame, Index, Series
 
-DATAFRAME_TYPES = {DataFrame}
-SERIES_TYPES = {Series}
-INDEX_TYPES = {Index}
+DATAFRAME_TYPES: Set[Type] = {DataFrame}
+SERIES_TYPES: Set[Type] = {Series}
+INDEX_TYPES: Set[Type] = {Index}
 
 if dask.DASK_INSTALLED:
     DATAFRAME_TYPES.update({dask.DataFrame})