Skip to content

Commit

Permalink
Add missing mypy types
Browse files Browse the repository at this point in the history
  • Loading branch information
camposandro committed Nov 1, 2024
1 parent d82bd17 commit 2836732
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/hats/io/file_io/file_io.py
Original file line number Diff line number Diff line change
Expand Up @@ -212,15 +212,15 @@ def write_parquet_metadata(
)


def read_fits_image(map_file_pointer: str | Path | UPath):
def read_fits_image(map_file_pointer: str | Path | UPath) -> np.ndarray:
"""Read the object spatial distribution information from a healpix FITS file.
Args:
map_file_pointer: location of file to be written
map_file_pointer (path-like): location of file to be written
Returns:
one-dimensional numpy array of integers where the value at each index corresponds
to the number of objects found at the healpix pixel.
one-dimensional numpy array of integers where the
value at each index corresponds to the number of objects found at the healpix pixel.
"""
map_file_pointer = get_upath(map_file_pointer)
return Skymap.from_fits(map_file_pointer).values
Expand Down

0 comments on commit 2836732

Please sign in to comment.