From 6f13a3a8bf0275a13901e8418fbf44c7c67610ed Mon Sep 17 00:00:00 2001
From: Gilad Turok <36947659+gil2rok@users.noreply.github.com>
Date: Wed, 30 Oct 2024 15:39:03 -0400
Subject: [PATCH] Update __init__.py

---
 blackjax/__init__.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/blackjax/__init__.py b/blackjax/__init__.py
index 5858c34aa..8b0b7be11 100644
--- a/blackjax/__init__.py
+++ b/blackjax/__init__.py
@@ -10,6 +10,7 @@
 from .adaptation.window_adaptation import window_adaptation
 from .base import SamplingAlgorithm, VIAlgorithm
 from .diagnostics import effective_sample_size as ess
+from .diagnostics import nested_rhat as nested_rhat
 from .diagnostics import potential_scale_reduction as rhat
 from .mcmc import barker
 from .mcmc import dynamic_hmc as _dynamic_hmc