MAINT: switch all cosmology to use wcosmo (!67) · Merge requests · RatesAndPopulations / GWPopulation Pipe

Colm Talbot requested to merge cosmology-fixes into master May 29, 2024

I had a few issues using the new cosmology functionality from !60 (merged) with the changes I made to use wcosmo.

make sure the correct model is used when using jax + cosmology
make sure constraints are used when using jax but not numpyro (I think this has been done somewhere else)
refactor the redshift/distance prior specification
make sure only the requested parameters are returned by the VTData
remove astropy dependency

I also set one of the CI tests to do cosmology fitting.

MAINT: switch all cosmology to use wcosmo