diff --git a/pyproject.toml b/pyproject.toml
index 676142b5b..03100dcf4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -63,7 +63,8 @@ cvxpy = [
 ]
 sbr = [
     "numpyro",
-    "jax"
+    "jax",
+    "arviz"
 ]
 
 [tool.black]