gpu runner

.github/workflows/notebooks.yaml

@@ -3,7 +3,7 @@ on: push
 
 jobs:
   run-notebooks:
-    runs-on: ubuntu-20.04
+    runs-on: 4-core-ubuntu-gpu-t4
     steps:
       - uses: actions/checkout@v2
       - uses: mamba-org/[email protected]
@@ -19,14 +19,8 @@ jobs:
       - name: Run notebooks
         shell: bash -l {0}
         run: |
-          set +e
+          set -e
           export PYTHONPATH=${PWD}:$PYTHONPATH
-          cd examples/
-          python -m flow_matching "import flow_matching; print('flow_matching module imported successfully')"
-          python -c "import flow_matching; print('flow_matching module imported successfully')"
-          python -c "import sys; print(sys.path)"
-          cd ..
-          
           for file in examples/*.ipynb; do
             jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
           done