test_numpyro_nuts() gives error when SN datafile changed

[kdesoto-astro]

File "/Users/jkubica/h/superphot-plus/src/superphot_plus/fit_numpyro.py", line 328, in run_mcmc res = mcmc.run( # pylint: disable=unused-variable File "/Users/jkubica/opt/anaconda3/envs/superphot/lib/python3.10/site-packages/numpyro/infer/mcmc.py", line 628, in run states_flat, last_state = partial_map_fn(map_args) File "/Users/jkubica/opt/anaconda3/envs/superphot/lib/python3.10/site-packages/numpyro/infer/mcmc.py", line 410, in _single_chain_mcmc new_init_state = self.sampler.init( File "/Users/jkubica/opt/anaconda3/envs/superphot/lib/python3.10/site-packages/numpyro/infer/hmc.py", line 713, in init init_params = self._init_state( File "/Users/jkubica/opt/anaconda3/envs/superphot/lib/python3.10/site-packages/numpyro/infer/hmc.py", line 657, in _init_state ) = initialize_model( File "/Users/jkubica/opt/anaconda3/envs/superphot/lib/python3.10/site-packages/numpyro/infer/util.py", line 745, in initialize_model raise RuntimeError( RuntimeError: Cannot find valid initial parameters. Please check your model again.

Happens when run on ZTF17aaaobyl.npz instead of default test file.

[jeremykubica]

Two quick updates:

• I was able to reproduce this bug with at least one other light curve (although I forgot to record which one).
• • The code seems to run fine with the SVI sampler.

[dougbrn]

I've also run into this when attempting to batch run a set of lightcurves with TAPE.