Test torch_geometric.llm in CI
#1788
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 RAG | |
| on: # yamllint disable-line rule:truthy | |
| push: | |
| branches: | |
| - master | |
| paths: | |
| - 'torch_geometric/datasets/web_qsp_dataset.py' | |
| - 'torch_geometric/llm/**' | |
| pull_request: | |
| paths: | |
| - 'torch_geometric/datasets/web_qsp_dataset.py' | |
| - 'torch_geometric/llm/**' | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number || github.ref }}-${{ startsWith(github.ref, 'refs/pull/') || github.run_number }} # yamllint disable-line | |
| # Only cancel intermediate builds if on a PR: | |
| cancel-in-progress: ${{ startsWith(github.ref, 'refs/pull/') }} | |
| jobs: | |
| rag_pytest: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - name: Checkout repository | |
| uses: actions/checkout@v5 | |
| - name: Setup packages | |
| uses: ./.github/actions/setup | |
| with: | |
| full_install: false | |
| - name: Install main package | |
| run: | | |
| uv pip install -e ".[test,rag]" | |
| - name: Run tests | |
| timeout-minutes: 10 | |
| run: | | |
| RAG_TEST=1 uv run --no-project pytest test/llm/models/test_g_retriever.py -m rag --cov --cov-report=xml -v | |
| RAG_TEST=1 uv run --no-project pytest test/llm/models/test_llm.py -m rag --cov --cov-report=xml -v | |
| RAG_TEST=1 uv run --no-project pytest test/llm/models/test_sentence_transformer.py -m rag --cov --cov-report=xml -v | |
| RAG_TEST=1 uv run --no-project pytest test/llm/utils/ -m rag --cov --cov-report=xml -v | |
| RAG_TEST=1 uv run --no-project pytest test/llm/test_rag_loader.py -m rag --cov --cov-report=xml -v | |
| RAG_TEST=1 uv run --no-project pytest test/datasets -m rag --cov --cov-report=xml -v | |
| shell: bash |