Add softmax notebook to be tested on CI. nbmake allows to run the notebook with pytest.

Author: James Choi

.github/workflows/test.yml

@@ -135,3 +135,11 @@ jobs:
           # -rf: print failed tests
           # --timeout: max allowed time for each test
           pytest -rf --timeout=60
+
+      - name: Run Notebook Tests
+        run: |
+          source .venv/bin/activate
+          # Run notebooks with longer timeout (notebooks include autotuning which takes time)
+          # --nbmake: execute notebooks and check for errors
+          # --overwrite: allow nbmake to overwrite notebook outputs
+          pytest --nbmake --overwrite notebooks/ --timeout=300 -v

requirements.txt

@@ -5,3 +5,4 @@ filecheck
 expecttest
 numpy
 rich
+nbmake