[ ] I have described the situation in which the bug arose, including what code was executed, information about my environment, and any applicable data others will need to reproduce the problem.
[ ] I have included available evidence of the unexpected behavior (including error messages, screenshots, and/or plots) as well as a descriprion of what I expected instead.
[ ] If I have a solution in mind, I have provided an explanation and/or pseudocode and/or task list.
Bug report
Some of SN Ia utility functions return float, while some others return a single-element 1-d array. So in the SN Ia end2end test,
source.get_param(state, parname)returnsfloatforparname"t0", "c", "redshift", "ra", "dec" and single-element 1-d array for "x1", "x2". https://github.com/lincc-frameworks/tdastro/blob/bcd50785bc63527bc8769a51f4eff17e0701f051/tests/tdastro/sources/test_snia.py#L167Before submitting Please check the following: