BUG: various packages missing from dev dependencies

[edschofield]

Describe the issue:

Following these docs on the contributing page:

conda env create -f conda-envs/environment-dev.yml

and attempting to run the test suite results in ModuleNotFoundErrors due to these modules being missing:

• numpyro: invoked in tests/sampling/test_jax.py
• blackjax: also invoked in tests/sampling/test_jax.py
• graphviz: also invoked in tests/test_model_graph.py

This also occurs when using a venv rather than a conda environment because requirements-dev.txt doesn't include these packages.

Reproduceable code example:

python -m pytest

Error message:

ModuleNotFoundError: No module named 'numpyro'
...
ModuleNotFoundError: No module named 'blackjax'
...
ModuleNotFoundError: No module named 'graphviz'

PyMC version information:

PyMC version from Git commit 340e403b8813ab5f3699a476cc828cc92c4f9d50.

Context for the issue:

These packages are required by the test suite...

There are several other files in conda-envs:

• environment-alternative-backends.yml
• environment-test.yml
• etc.

These aren't mentioned in the Contributor Docs and currently have no equivalents as requirements-*.txt for non-conda installs with pip or uv.

[ricardoV94]

numpyro tests should skip gracefully with pytest importorskip