Skip to content

Test torch_geometric.llm in CI #1789

Test torch_geometric.llm in CI

Test torch_geometric.llm in CI #1789

Workflow file for this run

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