From 613490526b802da261af34809a104990aeb353f0 Mon Sep 17 00:00:00 2001 From: Fengler Date: Wed, 8 May 2024 20:10:00 +0200 Subject: [PATCH] work on suggestions --- .github/workflows/run_tests.yml | 2 +- src/hssm/hssm.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml index 3061862d..5bd038f0 100644 --- a/.github/workflows/run_tests.yml +++ b/.github/workflows/run_tests.yml @@ -30,7 +30,7 @@ jobs: - name: install pymc and poetry run: | mamba info - mamba install -c conda-forge pymc=5.14 poetry + mamba install -c conda-forge pymc poetry - name: install hssm run: | diff --git a/src/hssm/hssm.py b/src/hssm/hssm.py index 6eb14ce4..97c520cf 100644 --- a/src/hssm/hssm.py +++ b/src/hssm/hssm.py @@ -502,7 +502,7 @@ def sample( # in the GLM as far as bambi is concerned) if self._inference_obj is not None: if self._parent not in self._inference_obj.posterior.data_vars.keys(): - self.sample_posterior_predictive(self._inference_obj, kind="mean") + self.model.predict(self._inference_obj, kind="mean", inplace=True) # rename 'rt,response_mean' to 'v' so in the traces everything # looks the way it should self._inference_obj.rename_vars(