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: Run notebooks
        shell: bash -l {0}
        env:
          PYTHONPATH: $PYTHONPATH:$PWD
        run: |
          set +e
          
          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
            PYTHONPATH=$PYTHONPATH jupyter nbconvert --to notebook --execute --ExecutePreprocessor.timeout=300 "$file"
          done