improve docstring

bambinos · Aug 12, 2022 · a43ad64 · a43ad64
1 parent 878d8c6
commit a43ad64
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bambi/models.py b/bambi/models.py
@@ -248,7 +248,8 @@ def fit(
 
         Returns
         -------
-        An ArviZ ``InferenceData`` instance if inference_method  ``"mcmc"`` (default).
+        An ArviZ ``InferenceData`` instance if inference_method  ``"mcmc"`` (default),
+        "nuts_numpyro" or "nuts_blackjax".
         An ``Approximation`` object if  ``"vi"`` and a dictionary if  ``"laplace"``.
         """
         method = kwargs.pop("method", None)