searchsorted strategy in amclouds.compute_cloud_base_pressure_index does not work. I need to think about the implementation of clouds.
rg_layer, MMRc = amp_nh3.calc_ammodel(
File "/home/kawahara/exojax/src/exojax/atm/atmphys.py", line 111, in calc_ammodel
pressure_base = pressures[ibase]
jax.errors.TracerArrayConversionError: The numpy.ndarray conversion method __array__() was called on traced array with shape int32[].
The error occurred while tracing the function update at /home/kawahara/anaconda3/lib/python3.10/site-packages/jaxopt/_src/optax_wrapper.py:121 for jit. This value became a tracer due to JAX operations on these lines:
searchsorted
strategy inamclouds.compute_cloud_base_pressure_index
does not work. I need to think about the implementation of clouds.