`test_histogram_approximation` failing due to warning in newer JAX release

[ricardoV94]

https://github.com/pymc-devs/pymc-experimental/actions/runs/7842985013/job/21402357791?pr=304#step:7:2388

[jessegrabowski]

Appears to be due to jax-ml/jax#18989

[ricardoV94]

Is the histogram functionality using JAX @ferrine? I can't find any evidence for that, but the JAX warning comes from the histogram tests

[jessegrabowski]

I think it's related to pm.sample, not histogram. There are test in test_linear_model and test_model_builder that are also failing because they call pm.sample.

We maybe have to split up the CI so that JAX dependencies are run separately and in JAX mode, like PyMC does

[ricardoV94]

So if JAX is loaded it will issue a warning on os.fork even if it's not being used?

[ricardoV94]

Then let's just ignore the warning, it's stupid