move cudf_accessor.py

unionai-oss · Jan 27, 2023 · aee1811 · aee1811
1 parent 2873068
commit aee1811
Show file tree

Hide file tree

Showing 5 changed files with 13 additions and 7 deletions.
diff --git a/pandera/cudf_accessor.py → pandera/accessors/cudf_accessor.py b/pandera/cudf_accessor.py → pandera/accessors/cudf_accessor.py
diff --git a/pandera/core/extensions.py b/pandera/core/extensions.py
@@ -3,7 +3,7 @@
 import warnings
 from enum import Enum
 from functools import partial, wraps
-from inspect import signature, Parameter, Signature, _empty
+from inspect import signature, Parameter, Signature, _empty  # type: ignore
 from typing import Any, Callable, Dict, List, Optional, Tuple, Type, Union
 
 import pandas as pd

diff --git a/pandera/typing/common.py b/pandera/typing/common.py
@@ -95,7 +95,7 @@
 else:
     GenericDtype = TypeVar(  # type: ignore
         "GenericDtype",
-        bound=Union[
+        bound=Union[  # type: ignore
             bool,
             int,
             str,
@@ -134,7 +134,7 @@
         ],
     )
 
-DataFrameModel = TypeVar("Schema", bound="DataFrameModel")  # type: ignore
+DataFrameModel = TypeVar("DataFrameModel", bound="DataFrameModel")  # type: ignore
 
 
 # pylint:disable=invalid-name

diff --git a/pandera/typing/cudf.py b/pandera/typing/cudf.py
@@ -6,8 +6,14 @@
 
 import pandas as pd
 
-from ..errors import SchemaError, SchemaInitError
-from .common import DataFrameBase, GenericDtype, IndexBase, Schema, SeriesBase
+from pandera.errors import SchemaError, SchemaInitError
+from pandera.typing.common import (
+    DataFrameBase,
+    GenericDtype,
+    IndexBase,
+    DataFrameModel,
+    SeriesBase,
+)
 from .formats import Formats
 
 try:
@@ -50,7 +56,7 @@ def __class_getitem__(cls, item):
     if TYPE_CHECKING:
         T = TypeVar("T")  # pragma: no cover
     else:
-        T = Schema
+        T = DataFrameModel
 
     # pylint:disable=too-few-public-methods
     class DataFrame(DataFrameBase, cudf.DataFrame, Generic[T]):

diff --git a/tests/cudf/test_cudf_accessor.py b/tests/cudf/test_cudf_accessor.py
@@ -3,7 +3,7 @@
 
 import pytest
 
-from pandera import cudf_accessor
+from pandera.accessors import cudf_accessor
 
 
 # pylint: disable=too-few-public-methods