notebooks.yaml

name: Notebooks
on: push

jobs:
  run-notebooks:
    runs-on: ubuntu-20.04
    steps:
      - uses: actions/checkout@v2
      - uses: mamba-org/setup-micromamba@v1.8.1
        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
        shell: bash -l {0}
        working-directory: examples
        run: |
          PYTHONPATH=../
          python -c "import sys; print(sys.path)"
          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