Track divergences in HMC stats

jessegrabowski · jessegrabowski · commit a27c9a45e77d · 2025-07-15T22:39:47.000+08:00
diff --git a/tests/step_methods/hmc/test_nuts.py b/tests/step_methods/hmc/test_nuts.py
@@ -148,6 +148,7 @@ def test_sampler_stats(self):
         expected_stat_names = {
             "depth",
             "diverging",
+            "divergences",
             "energy",
             "energy_error",
             "model_logp",
