diff --git a/tests/pandas/test_indexes.py b/tests/pandas/test_indexes.py index 4b60b0486..bae39aaf3 100644 --- a/tests/pandas/test_indexes.py +++ b/tests/pandas/test_indexes.py @@ -62,5 +62,6 @@ def test_column_contains() -> None: def test_difference_none() -> None: + # https://github.com/pandas-dev/pandas-stubs/issues/17 ind = pd.Index([1, 2, 3]) id = ind.difference([1, None]) diff --git a/typings/pandas/_typing.pyi b/typings/pandas/_typing.pyi index 30b2bc7ac..24362222c 100644 --- a/typings/pandas/_typing.pyi +++ b/typings/pandas/_typing.pyi @@ -80,7 +80,7 @@ Label = Optional[Hashable] Level = Union[Hashable, int] Ordered = Optional[bool] JSONSerializable = Union[PythonScalar, List, Dict] -Axes = Collection +Axes = Union[AnyArrayLike, List, Dict, range, Sequence[str]] Renamer = Union[Mapping[Any, Label], Callable[[Any], Label]] T = TypeVar("T") FuncType = Callable[..., Any] diff --git a/typings/pandas/core/frame.pyi b/typings/pandas/core/frame.pyi index 3f28a38bc..090e4bc28 100644 --- a/typings/pandas/core/frame.pyi +++ b/typings/pandas/core/frame.pyi @@ -65,8 +65,6 @@ from typing import ( Pattern, ) - - import numpy as _np import datetime as _dt @@ -178,8 +176,8 @@ class DataFrame(NDFrame, OpsMixin): def __new__( cls, data: Optional[Union[_ListLike, DataFrame, Dict[Any, Any]]] = ..., - index: Optional[Union[Index, _ListLike]] = ..., - columns: Optional[_ListLike] = ..., + index: Optional[Axes] = ..., + columns: Optional[Axes] = ..., dtype=..., copy: _bool = ..., ) -> DataFrame: ...