call convergence_check after sampling with numpyro

jessegrabowski · jessegrabowski · commit 3d370ff7085d · 2023-12-06T19:58:30.000+01:00
diff --git a/pymc/sampling/mcmc.py b/pymc/sampling/mcmc.py
@@ -359,6 +359,10 @@ def _sample_external_nuts(
             idata_kwargs=idata_kwargs,
             **nuts_sampler_kwargs,
         )
+
+        warns = run_convergence_checks(idata, model)
+        log_warnings(warns)
+
         return idata
 
     elif sampler == "blackjax":
