add compile_logp convenience function (#607)

* add compile_logp convenience function

* add convenience logp constructor from jax callable

* towards building hssm models from simulator and jax callable

* add tutorials

* small compile_logp test

* entering mypy hell

* reduce rv to one constructor

* update versions, run through tutorials

.github/workflows/run_fast_tests.yml

@@ -0,0 +1,36 @@
+name: Run fast tests
+
+on:
+  pull_request:
+  workflow_call:
+
+jobs:
+  run_tests:
+    runs-on: ubuntu-latest
+    if: ${{ ! contains(github.event.head_commit.message, '[skip fast tests]') }}
+    env:
+      PYTENSOR_FLAGS: "blas__ldflags=-L/usr/lib/x86_64-linux-gnu -lblas -llapack"
+
+    strategy:
+      fail-fast: true
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Setup Poetry environment
+        uses: ./.github/setup-poetry-env
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Run fast tests
+        run: poetry run pytest -n auto -s --ignore=tests/slow
+        env:
+          PYTENSOR_FLAGS: ${{ env.PYTENSOR_FLAGS }}
+
+      - name: Run jupyter notebooks in docs
+        run: poetry run mkdocs build
+        env:
+          PYTENSOR_FLAGS: ${{ env.PYTENSOR_FLAGS }}

.github/workflows/run_slow_tests.yml

@@ -0,0 +1,35 @@
+name: Run slow tests
+
+on:
+  pull_request:
+  workflow_call:
+
+jobs:
+  run_tests:
+    runs-on: ubuntu-latest
+    if: ${{ ! contains(github.event.head_commit.message, '[skip slow tests]') }}
+    env:
+      PYTENSOR_FLAGS: "blas__ldflags=-L/usr/lib/x86_64-linux-gnu -lblas -llapack"
+
+    strategy:
+      fail-fast: true
+      matrix:
+        python-version: ["3.10", "3.11", "3.12"]
+
+    steps:
+      - name: Checkout repository
+        uses: actions/checkout@v4
+
+      - name: Setup Poetry environment
+        uses: ./.github/setup-poetry-env
+        with:
+          python-version: ${{ matrix.python-version }}
+
+      - name: Install hssm
+        if: steps.cache.outputs.cache-hit != 'true'
+        run: poetry install --no-interaction
+
+      - name: Run slow tests
+        run: poetry run pytest -s tests/slow
+        env:
+          PYTENSOR_FLAGS: ${{ env.PYTENSOR_FLAGS }}

docs/changelog.md

@@ -2,6 +2,16 @@
 
 ## 0.2.x
 
+### 0.2.5
+
+This version of HSSM incorporated the following changes:
+
+1. We added a new tutorial on how to use custom likelihood functions with HSSM.
+2. Added convenience function `compile_logp()` to return a compiled log-likelihood function to be used freely downstream.
+3. Some improvements to internals to allow for (1.)
+4. Major improvments to plotting with `model_cartoon_plots()`
+5. Refactoring and minor improvements to pre-existing plots
+
 ### 0.2.4
 
 This version of HSSM incorporated the following changes:

docs/overrides/main.html

@@ -5,7 +5,7 @@
   </span>
   Navigate the site here!
 </span>
-<span class="right-margin"> v0.2.4 is released! </span>
+<span class="right-margin"> v0.2.5 is released! </span>
 <span>
   <span class="twemoji">
     {% include ".icons/material/head-question.svg" %}