separate action for notebooks

.github/workflows/ci.yaml

@@ -31,16 +31,6 @@ jobs:
         run: |
           coverage run --include='flow_matching/**/*.py' -m unittest discover tests -v
 
-      - name: Run notebooks
-        shell: bash -l {0}
-        working-directory: examples
-        run: |
-          set -e
-          PYTHONPATH=../
-          for file in ./*.ipynb; do
-            jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
-          done
-
       - name: Docstring Lint
         shell: bash -l {0}
         run: |

.github/workflows/notebooks.yaml

@@ -0,0 +1,31 @@
+name: Notebooks
+on: push
+
+jobs:
+  run-notebooks:
+    runs-on: ubuntu-20.04
+    steps:
+      - uses: actions/checkout@v2
+      - uses: mamba-org/[email protected]
+        with:
+          environment-file: environment.yml
+          cache-environment: true
+
+      - name: Check GPU availability
+        shell: bash -l {0}
+        run: |
+          python -c "import torch; print('Is CUDA available:', torch.cuda.is_available())"
+
+      - name: Import flow_matching
+        run: |
+          python -c "import flow_matching; print('flow_matching module imported successfully')"
+
+      - name: Run notebooks
+        shell: bash -l {0}
+        working-directory: examples
+        run: |
+          set -e
+          PYTHONPATH=../
+          for file in ./*.ipynb; do
+            jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
+          done