From 4906aa75d4270952a130a96d234f1264881868d7 Mon Sep 17 00:00:00 2001
From: Gibraan Rahman <grahman@eng.ucsd.edu>
Date: Wed, 31 Mar 2021 07:59:09 -0700
Subject: [PATCH] fix #21

---
 birdman/model_util.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/birdman/model_util.py b/birdman/model_util.py
index 46962aa..fdde7c4 100644
--- 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