TST fix CI tests after rebase on main

[beckermr]

This PR fixes the CI tests after the galsim branch was rebased on main.

[beckermr]

I'm working the merge with main here.

[ismael-mendoza]

The tests take significantly longer, is that expected @beckermr ?

[beckermr]

OK @ismael-mendoza this one is ready for review!

[beckermr]

I am not sure. How much longer? A few minutes?

[beckermr]

Looks like ~2x! See the comment below for why.

[beckermr]

OK I went through a bunch of old test suites. As far as I can tell, there is no big regression here. As we've added features, more of the galsim test suite is running and so things go longer. Here are some line items

• jax_galsim.random - the tests for these classes added ~2 minutes to the test suite
• jax_galsim.Deconvolve - once the tests for this were working, they added ~1 minute to the test suite
• jax_galsim.Image.wrap - the tests for this method added ~1 minute to the test suite
• tests/jax/test_api.py - these tests added ~1 minute to the test suite

Those alone accoutn for ~half the time.

[beckermr]

This one is ready to merge @ismael-mendoza!