wip: Fix typing

geoarrow · Mar 5, 2024 · 990c1ed · 990c1ed
1 parent 0e4473e
commit 990c1ed
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/python/core/python/geoarrow/rust/core/_rust.pyi b/python/core/python/geoarrow/rust/core/_rust.pyi
@@ -1201,13 +1201,22 @@ def envelope(input: ArrowStreamExportable) -> ChunkedRectArray: ...
 def envelope(
     input: ArrowArrayExportable | ArrowStreamExportable,
 ) -> RectArray | ChunkedRectArray: ...
+
+BroadcastGeometry = Union[ScalarGeometry, GeometryArray, ChunkedGeometryArray]
+
 @overload
-def frechet_distance(input: ArrowArrayExportable) -> Float64Array: ...
+def frechet_distance(
+    input: ArrowArrayExportable, other: BroadcastGeometry
+) -> Float64Array: ...
 @overload
-def frechet_distance(input: ArrowStreamExportable) -> ChunkedFloat64Array: ...
+def frechet_distance(
+    input: ArrowStreamExportable, other: BroadcastGeometry
+) -> ChunkedFloat64Array: ...
 def frechet_distance(
     input: ArrowArrayExportable | ArrowStreamExportable,
 ) -> Float64Array | ChunkedFloat64Array: ...
+
+
 def geodesic_perimeter(input: ArrowArrayExportable) -> Float64Array: ...
 @overload
 def is_empty(input: ArrowArrayExportable) -> BooleanArray: ...