Unverified Commit c122d065 authored by Akke Viitanen's avatar Akke Viitanen
Browse files

print arguments while reading fits files

parent f2b6197e
Loading
Loading
Loading
Loading
+5 −4
Original line number Diff line number Diff line
@@ -774,7 +774,7 @@ def get_stellar_mass_completeness_cosmos2020(type: str, redshift: float) -> floa
    return f1 * (1 + redshift) + f2 * (1 + redshift) ** 2


def read_fits(filename, *args, **kwargs):
def read_fits(filename: str, *args, **kwargs):
    """
    Read a FITS filename with supressed error messages.
    """
@@ -782,7 +782,7 @@ def read_fits(filename, *args, **kwargs):

    from astropy.units import UnitsWarning

    logger.info(f"Reading {filename}")
    logger.info(f"Reading {filename} {args=} {kwargs=}")
    ret = None
    with warnings.catch_warnings():
        warnings.simplefilter("ignore", UnitsWarning)
@@ -790,7 +790,7 @@ def read_fits(filename, *args, **kwargs):
    return ret


def read_table(filename):
def read_table(filename: str, *args, **kwargs):
    """
    Read an astropy table but do so silently.
    """
@@ -799,7 +799,8 @@ def read_table(filename):
    from astropy.table import Table
    from astropy.units import UnitsWarning

    logger.info(f"Reading {filename} {args=} {kwargs=}")
    with warnings.catch_warnings():
        warnings.simplefilter("ignore", UnitsWarning)
        table = Table.read(filename)
        table = Table.read(filename, *args, **kwargs)
    return table