TXT2KG w/ txtkg_rag.py e2e workflow example
#91
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Testing distributed PyG | |
| on: # yamllint disable-line rule:truthy | |
| push: | |
| branches: | |
| - master | |
| pull_request: | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ startsWith(github.ref, 'refs/pull/') || github.run_number }} | |
| # Only cancel intermediate builds if on a PR: | |
| cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }} | |
| jobs: | |
| dist_pytest: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v4 | |
| with: | |
| fetch-depth: 40 | |
| # Skip workflow if only certain files have been changed. | |
| - name: Get changed files | |
| id: changed-files-specific | |
| uses: tj-actions/changed-files@v46 | |
| with: | |
| files: | | |
| torch_geometric/distributed/** | |
| test/distributed/** | |
| - name: Setup packages | |
| if: steps.changed-files-specific.outputs.any_changed == 'true' | |
| uses: ./.github/actions/setup | |
| - name: Install main package | |
| if: steps.changed-files-specific.outputs.any_changed == 'true' | |
| run: | | |
| pip install -e ".[test]" | |
| python -c "import torch; print('PyTorch:', torch.__version__)" | |
| python -c "import torch; print('CUDA available:', torch.cuda.is_available())" | |
| python -c "import torch; print('CUDA:', torch.version.cuda)" | |
| - name: Run tests | |
| if: steps.changed-files-specific.outputs.any_changed == 'true' | |
| timeout-minutes: 10 | |
| run: | | |
| DIST_TEST=1 pytest test/distributed | |
| shell: bash |