diff --git a/.github/workflows/notebooks.yaml b/.github/workflows/notebooks.yaml index fd3d416..877ee26 100644 --- a/.github/workflows/notebooks.yaml +++ b/.github/workflows/notebooks.yaml @@ -17,6 +17,8 @@ jobs: python -c "import torch; print('Is CUDA available:', torch.cuda.is_available())" - name: Import flow_matching + shell: bash -l {0} + working-directory: examples run: | python -c "import flow_matching; print('flow_matching module imported successfully')"