From 2fb26364fc43f758e3826acea11a6f2d341533af Mon Sep 17 00:00:00 2001 From: Ed Schofield Date: Fri, 26 Sep 2025 14:27:55 +0200 Subject: [PATCH 1/3] Add numpyro to conda-envs/environment-dev.yml --- conda-envs/environment-dev.yml | 1 + 1 file changed, 1 insertion(+) diff --git a/conda-envs/environment-dev.yml b/conda-envs/environment-dev.yml index b6a9c52b91..4d5cd4361c 100644 --- a/conda-envs/environment-dev.yml +++ b/conda-envs/environment-dev.yml @@ -25,6 +25,7 @@ dependencies: - jupyter-sphinx - myst-nb<=1.0.0 - numpydoc +- numpyro - pre-commit>=2.8.0 - polyagamma - pytest-cov>=2.5 From 12eab37ecd5e43a9f60630806b18d0424b17f7d9 Mon Sep 17 00:00:00 2001 From: Ed Schofield Date: Fri, 26 Sep 2025 14:40:55 +0200 Subject: [PATCH 2/3] Add numpyro to requirements-dev.yml --- requirements-dev.txt | 1 + 1 file changed, 1 insertion(+) diff --git a/requirements-dev.txt b/requirements-dev.txt index 8b59683594..70afae7ce3 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -12,6 +12,7 @@ myst-nb<=1.0.0 numdifftools>=0.9.40 numpy>=1.25.0 numpydoc +numpyro pandas>=0.24.0 polyagamma pre-commit>=2.8.0 From 72261d68763cfeb2cbb047295ce9a0c41bb16655 Mon Sep 17 00:00:00 2001 From: Ed Schofield Date: Fri, 26 Sep 2025 15:59:33 +0200 Subject: [PATCH 3/3] Add blackjax as a dev depeendency --- conda-envs/environment-dev.yml | 1 + requirements-dev.txt | 1 + 2 files changed, 2 insertions(+) diff --git a/conda-envs/environment-dev.yml b/conda-envs/environment-dev.yml index 4d5cd4361c..22ba03f95d 100644 --- a/conda-envs/environment-dev.yml +++ b/conda-envs/environment-dev.yml @@ -20,6 +20,7 @@ dependencies: - threadpoolctl>=3.1.0 - zarr>=2.5.0,<3 # Extra dependencies for dev, testing and docs build +- blackjax - ipython>=7.16 - jax - jupyter-sphinx diff --git a/requirements-dev.txt b/requirements-dev.txt index 70afae7ce3..bcd20f1fd7 100644 --- a/requirements-dev.txt +++ b/requirements-dev.txt @@ -2,6 +2,7 @@ # See that file for comments about the need/usage of each dependency. arviz>=0.13.0 +blackjax cachetools>=4.2.1 cloudpickle ipython>=7.16