tests.yml

# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

name: CI

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

jobs:

  style:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - name: Set up Python
        uses: astral-sh/setup-uv@v5
        with:
          python-version: "3.11"

      - name: Install dependencies
        run: uv pip install -r requirements/test.txt

      # Update output format to enable automatic inline annotations.
      - name: Run Ruff Linter
        run: ruff check --output-format=github
      - name: Run Ruff Formatter
        run: ruff format --check

  test:
    strategy:
      matrix:
        os: [ubuntu-latest, macos-latest, windows-latest]
        python-version: ["3.10", "3.11", "3.12"]
    runs-on: ${{ matrix.os }}
    steps:
    - uses: actions/checkout@v4

    - name: Set up Python ${{ matrix.python-version }}
      uses: astral-sh/setup-uv@v5
      with:
        python-version: ${{ matrix.python-version }}

    - name: Install dependencies
      run: uv pip install -r requirements/required.txt -r requirements/test.txt

    - name: Show installed packages
      run: uv pip list

    - name: Test with PyTest
      run: uv run pytest -v -rsx -n 2 --cov=segmentation_models_pytorch --cov-report=xml --cov-config=pyproject.toml --non-marked-only

    - name: Upload coverage reports to Codecov
      uses: codecov/codecov-action@v5
      with:
        token: ${{ secrets.CODECOV_TOKEN }}
        slug: qubvel-org/segmentation_models.pytorch
      if: matrix.os == 'macos-latest' && matrix.python-version == '3.12'

  test_logits_match:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: astral-sh/setup-uv@v5
      with:
        python-version: "3.10"
    - name: Install dependencies
      run: uv pip install -r requirements/required.txt -r requirements/test.txt
    - name: Show installed packages
      run: uv pip list
    - name: Test with PyTest
      run: RUN_SLOW=1 uv run pytest -v -rsx -n 2 -m "logits_match"

  test_torch_compile:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: astral-sh/setup-uv@v5
      with:
        python-version: "3.10"
    - name: Install dependencies
      run: uv pip install -r requirements/required.txt -r requirements/test.txt
    - name: Show installed packages
      run: uv pip list
    - name: Test with PyTest
      run: uv run pytest -v -rsx -n 2 -m "compile"

  test_torch_export:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: astral-sh/setup-uv@v5
      with:
        python-version: "3.10"
    - name: Install dependencies
      run: uv pip install -r requirements/required.txt -r requirements/test.txt
    - name: Show installed packages
      run: uv pip list
    - name: Test with PyTest
      run: uv run pytest -v -rsx -n 2 -m "torch_export"

  test_torch_script:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: astral-sh/setup-uv@v5
      with:
        python-version: "3.10"
    - name: Install dependencies
      run: uv pip install -r requirements/required.txt -r requirements/test.txt
    - name: Show installed packages
      run: uv pip list
    - name: Test with PyTest
      run: uv run pytest -v -rsx -n 2 -m "torch_script"

  minimum:
    runs-on: ubuntu-latest
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: astral-sh/setup-uv@v5
      with:
        python-version: "3.9"
    - name: Install dependencies
      run: uv pip install -r requirements/minimum.old -r requirements/test.txt
    - name: Show installed packages
      run: uv pip list
    - name: Test with pytest
      run: uv run pytest -v -rsx -n 2 --non-marked-only