update ssm-simulators dependency and allow 3.12 tests

.github/workflows/run_fast_tests.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       fail-fast: true
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
       - name: Checkout repository

.github/workflows/run_slow_tests.yml

@@ -13,7 +13,7 @@ jobs:
     strategy:
       fail-fast: true
       matrix:
-        python-version: ["3.10", "3.11"]
+        python-version: ["3.10", "3.11", "3.12"]
 
     steps:
       - name: Checkout repository

pyproject.toml

@@ -15,11 +15,11 @@ repository = "https://github.com/lnccbrown/HSSM"
 keywords = ["HSSM", "sequential sampling models", "bayesian", "bayes", "mcmc"]
 
 [tool.poetry.dependencies]
-python = ">=3.10,<3.12"
+python = ">=3.10,<=3.12"
 pymc = ">=5.16.2,<5.17.0"
 arviz = "^0.19.0"
 onnx = "^1.16.0"
-ssm-simulators = "^0.7.2"
+ssm-simulators = "^0.7.5"
 huggingface-hub = "^0.24.6"
 bambi = ">=0.14.0,<0.15.0"
 numpyro = "^0.15.2"

src/hssm/hssm.py

@@ -651,8 +651,7 @@ def sample(
             self.log_likelihood(self._inference_obj, inplace=True)
 
         # Subset data vars in posterior
-        if self._inference_obj is not None:
-            self._clean_posterior_group(idata=self._inference_obj)
+        self._clean_posterior_group(idata=self._inference_obj)
         return self.traces
 
     def vi(
@@ -708,8 +707,7 @@ def vi(
             self._inference_obj_vi = self._vi_approx.sample(draws)
 
         # Post-processing
-        if self._inference_obj_vi is not None:
-            self._clean_posterior_group(idata=self._inference_obj_vi)
+        self._clean_posterior_group(idata=self._inference_obj_vi)
 
         # Return the InferenceData object if return_idata is True
         if return_idata: