Add jupytext for notebooks

.github/workflows/ci.yml

@@ -40,6 +40,40 @@ jobs:
       - name: Unit tests
         run: python -m pytest -v tests/unit_tests
 
+  notebook-tests:
+    name: Notebook Execution
+    runs-on: ubuntu-latest
+    env:
+      PUFFER_CPU: 1
+    steps:
+      - name: Checkout code
+        uses: actions/checkout@v4
+
+      - name: Setup Python
+        uses: actions/setup-python@v5
+        with:
+          python-version: "3.11"
+
+      - name: Free up disk space
+        run: |
+          sudo rm -rf /usr/share/dotnet /usr/local/lib/android /opt/ghc
+          sudo apt-get clean
+          sudo rm -rf ~/.cache/pip /tmp/* /var/tmp/*
+
+      - name: Install pufferlib
+        env:
+          TMPDIR: ${{ runner.temp }}/build
+          PIP_NO_CACHE_DIR: 1
+        run: |
+          sudo apt-get update && sudo apt-get install -y build-essential cmake
+          python -m pip install -U pip pytest jupytext nbclient ipykernel ipywidgets
+          pip install -e . --no-cache-dir
+          python setup.py build_ext --inplace --force
+
+      - name: Run notebooks
+        run: python -m pytest -v tests/notebooks
+        timeout-minutes: 30
+
   smoke-tests:
     name: Smoke Train Test
     runs-on: ubuntu-latest

.pre-commit-config.yaml

@@ -14,13 +14,6 @@ repos:
     # - id: ruff
     #   args: [ --fix ]
     - id: ruff-format
-# Cleaning Jupyter notebooks
-- repo: https://github.com/srstevenson/nb-clean
-  rev: 4.0.1
-  hooks:
-    - id: nb-clean
-      args:
-        - --remove-empty-cells
 # Apply clang format
 - repo: https://github.com/pre-commit/mirrors-clang-format
   rev: 'v21.1.5'

notebooks/.gitignore

@@ -0,0 +1 @@
+*.ipynb