diff --git a/src/hats/pixel_math/healpix_shim.py b/src/hats/pixel_math/healpix_shim.py index 96eb18b7..2f1d8a22 100644 --- a/src/hats/pixel_math/healpix_shim.py +++ b/src/hats/pixel_math/healpix_shim.py @@ -58,14 +58,14 @@ def order2pixarea(order: int, degrees: bool = False) -> float: return pix_area_rad -def radec2pix(order: int, ra: float, dec: float) -> int: +def radec2pix(order: int, ra: float, dec: float) -> np.ndarray[np.int64]: if not is_order_valid(order): raise ValueError("Invalid value for order") ra = Longitude(ra, unit="deg") dec = Latitude(dec, unit="deg") - return cdshealpix.lonlat_to_healpix(ra, dec, order) + return cdshealpix.lonlat_to_healpix(ra, dec, order).astype(np.int64) def ring2nest(order: int, ipix: int) -> int: @@ -92,15 +92,6 @@ def ang2vec(ra, dec, **kwargs) -> np.ndarray: return np.array([coords.x.value, coords.y.value, coords.z.value]).T -## FITS -def read_map(*args, **kwargs): - return hp.read_map(*args, **kwargs) - - -def write_map(*args, **kwargs): - return hp.write_map(*args, **kwargs) - - ## Custom functions