From 6e35658d1e6fd0c8ad74cee4e04f5fe273e0ef06 Mon Sep 17 00:00:00 2001
From: Marton Havasi <marton@meta.com>
Date: Wed, 11 Dec 2024 22:06:28 +0000
Subject: [PATCH] gpu runner

---
 .github/workflows/notebooks.yaml | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/.github/workflows/notebooks.yaml b/.github/workflows/notebooks.yaml
index 5e86925..4bca260 100644
--- a/.github/workflows/notebooks.yaml
+++ b/.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/setup-micromamba@v1.8.1
@@ -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
\ No newline at end of file