Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion stubs/geopandas/METADATA.toml
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
version = "1.1.1"
# Requires a version of numpy with a `py.typed` file
requires = ["numpy>=1.20", "pandas-stubs<2.3.3.251201", "types-shapely", "pyproj"]
requires = ["numpy>=1.20", "pandas-stubs", "types-shapely", "pyproj"]
upstream_repository = "https://github.com/geopandas/geopandas"

[tool.stubtest]
Expand Down
21 changes: 7 additions & 14 deletions stubs/geopandas/geopandas/array.pyi
Original file line number Diff line number Diff line change
@@ -1,12 +1,13 @@
import builtins
from _typeshed import Incomplete, Unused
from collections.abc import Callable, Collection, Sequence
from collections.abc import Callable, Collection
from typing import Any, ClassVar, Final, Literal, NoReturn, SupportsIndex, TypeVar, overload
from typing_extensions import Self, TypeAlias, deprecated

import numpy as np
import pandas as pd
from numpy.typing import ArrayLike, DTypeLike, NDArray
from pandas._typing import ScalarIndexer, SequenceIndexer, TakeIndexer
from pandas.api.extensions import ExtensionArray, ExtensionDtype
from pyproj import CRS, Transformer
from shapely import Geometry
Expand Down Expand Up @@ -68,15 +69,9 @@ class GeometryArray(ExtensionArray):
def __len__(self) -> int: ...
# np.integer[Any] because precision is not important
@overload
def __getitem__(self, idx: int | np.integer[Any]) -> BaseGeometry: ... # Always 1-D, doesn't accept tuple
def __getitem__(self, idx: ScalarIndexer) -> BaseGeometry: ... # Always 1-D, doesn't accept tuple
@overload
def __getitem__(
self, idx: slice | Sequence[SupportsIndex] | NDArray[np.bool_] | NDArray[np.integer[Any]]
) -> GeometryArray: ...
@overload
def __getitem__(
self, idx: int | np.integer[Any] | slice | Sequence[int] | NDArray[np.bool_] | NDArray[np.integer[Any]]
) -> BaseGeometry | GeometryArray: ...
def __getitem__(self, idx: SequenceIndexer) -> GeometryArray: ...
def __setitem__(
self, key, value: _ArrayOrGeom | pd.DataFrame | pd.Series[Any] # Cannot use pd.Series[BaseGeometry]
) -> None: ...
Expand Down Expand Up @@ -222,17 +217,15 @@ class GeometryArray(ExtensionArray):
@property
def ndim(self) -> Literal[1]: ...
def copy(self, *args: Unused, **kwargs: Unused) -> GeometryArray: ...
def take(
self, indices: Sequence[SupportsIndex] | NDArray[np.integer], allow_fill: bool = False, fill_value: Geometry | None = None
) -> GeometryArray: ...
def fillna(
def take(self, indices: TakeIndexer, allow_fill: bool = False, fill_value: Geometry | None = None) -> GeometryArray: ...
def fillna( # type: ignore[override]
self,
value: Geometry | GeometryArray | None = None,
method: Literal["backfill", "bfill", "pad", "ffill"] | None = None,
limit: int | None = None,
copy: bool = True,
) -> GeometryArray: ...
@overload
@overload # type: ignore[override]
def astype(self, dtype: GeometryDtype, copy: bool = True) -> GeometryArray: ...
@overload
def astype(self, dtype: ExtensionDtype | Literal["string"], copy: bool = True) -> ExtensionArray: ... # type: ignore[overload-overlap]
Expand Down