fix #21

biocore · Mar 31, 2021 · 4906aa7 · 4906aa7
1 parent dcd55a5
commit 4906aa7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/birdman/model_util.py b/birdman/model_util.py
@@ -18,7 +18,7 @@ def single_fit_to_inference(
     posterior_predictive: str = None,
     log_likelihood: str = None,
     sample_names: Sequence[str] = None,
-) -> xr.Dataset:
+) -> az.InferenceData:
     """Convert fitted Stan model into inference object.
 
     :param fit: Fitted model