From a21ef9c416e9daae3c972bef5ddda96c163eeea1 Mon Sep 17 00:00:00 2001
From: Ali Hamdan <ali.hamdan.dev@gmail.com>
Date: Tue, 16 Sep 2025 15:25:56 +0200
Subject: [PATCH] shapely: fix array dtype in geometry functions return type

---
 stubs/shapely/shapely/_geometry.pyi     | 16 ++++++++--------
 stubs/shapely/shapely/_ragged_array.pyi |  2 +-
 2 files changed, 9 insertions(+), 9 deletions(-)

diff --git a/stubs/shapely/shapely/_geometry.pyi b/stubs/shapely/shapely/_geometry.pyi
index d2e2055bce02..3e1a92ca5ab0 100644
--- a/stubs/shapely/shapely/_geometry.pyi
+++ b/stubs/shapely/shapely/_geometry.pyi
@@ -54,23 +54,23 @@ class GeometryType(IntEnum):
 @overload
 def get_type_id(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_type_id(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_type_id(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_dimensions(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_dimensions(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_dimensions(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_coordinate_dimension(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_coordinate_dimension(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_coordinate_dimension(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_num_coordinates(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_num_coordinates(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_num_coordinates(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_srid(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_srid(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_srid(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def set_srid(geometry: OptGeoT, srid: SupportsIndex, **kwargs) -> OptGeoT: ...
 @overload
@@ -106,7 +106,7 @@ def get_point(geometry: OptGeoArrayLike, index: ArrayLikeSeq[SupportsIndex], **k
 @overload
 def get_num_points(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_num_points(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_num_points(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_exterior_ring(geometry: Polygon, **kwargs) -> LinearRing: ...
 @overload
@@ -128,7 +128,7 @@ def get_interior_ring(geometry: OptGeoArrayLike, index: ArrayLikeSeq[SupportsInd
 @overload
 def get_num_interior_rings(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_num_interior_rings(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_num_interior_rings(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_geometry(geometry: MultiPoint, index: SupportsIndex, **kwargs) -> Point | Any: ...
 @overload
@@ -160,7 +160,7 @@ def get_rings(geometry: OptGeoArrayLike, return_index: bool) -> GeoArray | tuple
 @overload
 def get_num_geometries(geometry: Geometry | None, **kwargs) -> np.int32: ...
 @overload
-def get_num_geometries(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int64]: ...
+def get_num_geometries(geometry: OptGeoArrayLikeSeq, **kwargs) -> NDArray[np.int32]: ...
 @overload
 def get_precision(geometry: Geometry | None, **kwargs) -> np.float64: ...
 @overload
diff --git a/stubs/shapely/shapely/_ragged_array.pyi b/stubs/shapely/shapely/_ragged_array.pyi
index c4d60165b8f5..a0822d55e5c6 100644
--- a/stubs/shapely/shapely/_ragged_array.pyi
+++ b/stubs/shapely/shapely/_ragged_array.pyi
@@ -6,7 +6,7 @@ from ._typing import ArrayLike, ArrayLikeSeq, GeoArray, OptGeoArrayLikeSeq
 
 def to_ragged_array(
     geometries: OptGeoArrayLikeSeq, include_z: bool | None = None, include_m: bool | None = None
-) -> tuple[GeometryType, NDArray[np.float64], tuple[NDArray[np.int64], ...]]: ...
+) -> tuple[GeometryType, NDArray[np.float64], tuple[NDArray[np.int32], ...]]: ...
 def from_ragged_array(
     geometry_type: GeometryType, coords: ArrayLike[float], offsets: ArrayLikeSeq[int] | None = None
 ) -> GeoArray: ...