Pointwise reranker for diver and reasoner-embed #653
Workflow file for this run
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
| on: | |
| pull_request: | |
| branches: | |
| - main | |
| jobs: | |
| lint: | |
| runs-on: ubuntu-latest | |
| env: | |
| TMPDIR: /mnt/tmp | |
| PRE_COMMIT_HOME: /mnt/pre-commit-cache | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Prepare temp/cache directories on /mnt | |
| run: | | |
| sudo mkdir -p "$TMPDIR" "$PRE_COMMIT_HOME" | |
| sudo chown "$USER":"$USER" "$TMPDIR" "$PRE_COMMIT_HOME" | |
| df -h / | |
| df -h /mnt | |
| - uses: astral-sh/setup-uv@v6 | |
| with: | |
| python-version: "3.11" | |
| - name: Create lint environment | |
| run: uv venv --python 3.11 | |
| - name: Install lint dependencies | |
| run: uv pip install --python .venv/bin/python pre-commit black isort | |
| - name: Run lint | |
| run: .venv/bin/pre-commit run --all-files | |
| unit_tests: | |
| needs: lint | |
| runs-on: ubuntu-latest | |
| strategy: | |
| matrix: | |
| include: | |
| - test-suite: analysis | |
| install-target: . | |
| - test-suite: evaluation | |
| install-target: ".[pyserini]" | |
| needs-java: true | |
| - test-suite: rerank | |
| install-target: ".[cloud,vllm]" | |
| env: | |
| TMPDIR: /mnt/tmp | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - name: Prepare temp directory on /mnt | |
| run: | | |
| sudo mkdir -p "$TMPDIR" | |
| sudo chown "$USER":"$USER" "$TMPDIR" | |
| df -h / | |
| df -h /mnt | |
| - uses: actions/setup-java@v3 | |
| if: ${{ matrix.needs-java }} | |
| with: | |
| distribution: "temurin" | |
| java-version: "21" | |
| - uses: astral-sh/setup-uv@v6 | |
| with: | |
| python-version: "3.11" | |
| - name: Create test environment | |
| run: uv venv --python 3.11 | |
| - name: Install test environment | |
| run: uv pip install --python .venv/bin/python -e "${{ matrix.install-target }}" | |
| - name: Run tests | |
| run: .venv/bin/python -m unittest discover -s "test/${{ matrix.test-suite }}" |