pipeline_tests.yml

name: Pipeline tests

on:
  push:
    branches: [main]
  pull_request:
    branches: [main]

env:
  ASPARAGUS_MODELS: "${{ github.workspace }}/models"
  #ASPARAGUS: "${{ github.workspace }}/tests/models"
  HYDRA_FULL_ERROR: 1

jobs:
  pipeline-tests:
    runs-on: ubuntu-latest
    timeout-minutes: 20

    steps:
      - uses: actions/checkout@v4

      - uses: actions/setup-python@v5
        with:
          python-version: "3.11"
          cache: pip

      - name: Install CPU-only torch then package
        run: |
          pip install torch torchvision --index-url https://download.pytorch.org/whl/cpu
          pip install -e ".[test]"

      - name: Run pretrain test
        run: |
          pytest tests/test_pretrain.py -v --timeout=300

      - name: Run finetune_seg test
        run: |
          pytest tests/test_finetune_seg.py -v --timeout=300

      - name: Run train_reg test
        run: |
          pytest tests/test_train_reg.py -v --timeout=300

      - name: Run test_cls test
        run: |
          pytest tests/test_test_cls.py -v --timeout=300

      - name: Run linear probe test
        run: |
          pytest tests/test_linear_probe.py -v --timeout=300