Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
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