Skip to content

Commit

Permalink
type hints
Browse files Browse the repository at this point in the history
  • Loading branch information
Huite committed Dec 11, 2023
1 parent ab64846 commit b8f187c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions numba_celltree/geometry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -142,7 +142,7 @@ def point_in_polygon(p: Point, poly: Sequence) -> bool:


@nb.njit(inline="always")
def in_bounds(p: Point, a: Point, b: Point):
def in_bounds(p: Point, a: Point, b: Point) -> bool:
"""
Check whether point p falls within the bounding box created by a and b
(after we've checked the size of the cross product).
Expand Down Expand Up @@ -246,7 +246,7 @@ def bounding_box(
def build_bboxes(
faces: IntArray,
vertices: FloatArray,
) -> Tuple[FloatArray, IntArray]:
) -> FloatArray:
# Make room for the bounding box of every polygon.
n_polys = len(faces)
bbox_coords = np.empty((n_polys, NDIM * 2), FloatDType)
Expand Down

0 comments on commit b8f187c

Please sign in to comment.