I think this should be `astropy.cosmology` - can you check what other sub-packages do?