shapely: fix array dtype in geometry functions return type (#14730)

Author: hamdanal

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

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: ...