diff --git a/python/fbprophet/models.py b/python/fbprophet/models.py index 59a741a..ca81631 100644 --- a/python/fbprophet/models.py +++ b/python/fbprophet/models.py @@ -140,7 +140,7 @@ class CmdStanPyBackend(IStanBackend): args.update(kwargs) self.stan_fit = self.model.sample(**args) - res = self.stan_fit.sample + res = self.stan_fit.draws() (samples, c, columns) = res.shape res = res.reshape((samples * c, columns)) params = self.stan_to_dict_numpy(self.stan_fit.column_names, res)