use numba jit compilation

Author: smcguire-cmu

src/hats/pixel_math/margin_bounding.py

@@ -90,7 +90,7 @@ def check_margin_bounds(r_asc, dec, pixel_order, pixel, margin_threshold, step=1
 
 
 # numba jit compiler doesn't count for coverage tests, so we'll set no cover.
-@njit("double(double[:], double[:], double)")
+@njit
 def _find_minimum_distance(separations, distances, margin_threshold):  # pragma: no cover
     """Find the minimum distance between a given datapoint and a healpixel"""
     minimum_index = np.argmin(separations)

src/hats/pixel_tree/moc_filter.py

@@ -26,12 +26,7 @@ def filter_by_moc(
     return PixelTree(tree.tree[tree_mask], tree.tree_order)
 
 
-@njit(
-    numba.bool_[::1](
-        numba.int64[:, :],
-        numba.uint64[:, :],
-    )
-)
+@njit
 def perform_filter_by_moc(
     tree: np.ndarray,
     moc: np.ndarray,

src/hats/pixel_tree/pixel_alignment.py

@@ -141,7 +141,7 @@ def get_pixel_mapping_df(mapping: np.ndarray, map_order: int) -> pd.DataFrame:
 
 
 # pylint: disable=too-many-statements
-@njit(numba.int64[::1, :](numba.int64[:, :], numba.int64[:, :]))
+@njit
 def perform_inner_align_trees(
     left: np.ndarray,
     right: np.ndarray,
@@ -208,15 +208,7 @@ def perform_inner_align_trees(
     return mapping[:out_index].T
 
 
-@njit(
-    numba.types.void(
-        numba.int64,
-        numba.int64,
-        numba.int64[:],
-        numba.boolean,
-        numba.types.List(numba.int64[::1]),
-    )
-)
+@njit
 def _add_pixels_until(
     add_from: int,
     add_to: int,
@@ -259,15 +251,7 @@ def _add_pixels_until(
         add_from = add_from + pixel_size
 
 
-@njit(
-    numba.types.void(
-        numba.int64,
-        numba.int64[:, :],
-        numba.int64,
-        numba.boolean,
-        numba.types.List(numba.int64[::1]),
-    )
-)
+@njit
 def _add_remaining_pixels(
     added_until: int,
     pixel_list: np.ndarray,
@@ -303,7 +287,7 @@ def _add_remaining_pixels(
 
 
 # pylint: disable=too-many-statements
-@njit(numba.types.List(numba.int64[::1])(numba.int64[:, :], numba.int64[:, :], numba.boolean, numba.boolean))
+@njit
 def perform_align_trees(
     left: np.ndarray,
     right: np.ndarray,