work on suggestions

lnccbrown · May 8, 2024 · 6134905 · 6134905
1 parent 3391ae4
commit 6134905
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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(