diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index 5451596c9..9ea250d68 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -148,6 +148,7 @@ jobs:
           XLA_FLAGS="--xla_force_host_platform_device_count=2" uv run pytest -vs test/contrib/test_tfp.py -k "chain"
           XLA_FLAGS="--xla_force_host_platform_device_count=2" uv run pytest -vs test/infer/test_hmc_gibbs.py -k "chain"
           XLA_FLAGS="--xla_force_host_platform_device_count=2" uv run pytest -vs test/infer/test_mcmc.py -k "chain or pmap or vmap"
+          XLA_FLAGS="--xla_force_host_platform_device_count=2" uv run pytest -vs test/test_compile.py -k "chain"
       - name: Test custom prng
         run: |
           JAX_ENABLE_CUSTOM_PRNG=1 uv run pytest -vs test/infer/test_mcmc.py
diff --git a/pyproject.toml b/pyproject.toml
index f50b70bad..23d1e687c 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -49,11 +49,19 @@ dev = [
     "jaxns>=2.6.3,<=2.6.9",
     "matplotlib",
     "optax>=0.0.6",
+    "pandas>=3.0.3",
     "pylab-sdk",                         # jaxns dependency
+    "pyro-api>=0.1.2",
+    "pytest>=9.1.1",
     "pytest-cov",
     "pyyaml",                            # flax dependency
     "requests",                          # pylab dependency
+    "ruff>=0.15.22",
+    "scikit-learn>=1.9.0",
+    "seaborn>=0.13.2",
     "tfp-nightly",
+    "ty>=0.0.57",
+    "wordcloud>=1.9.6",
 ]
 ci = [
     "coverage>=7.13.5",
diff --git a/test/test_compile.py b/test/test_compile.py
index 2971a1df1..1531c7f34 100644
--- a/test/test_compile.py
+++ b/test/test_compile.py
@@ -53,7 +53,11 @@ def test_mcmc_parallel_chain(deterministic):
     mcmc.get_samples()
 
     if deterministic:
-        assert GLOBAL["count"] == 4
+        # As in test_mcmc_one_chain, we have two extra calls to the model
+        # to get deterministic values:
+        # 1. transform the init state
+        # 2. transform state during the loop
+        assert GLOBAL["count"] == 5
     else:
         assert GLOBAL["count"] == 3