name: Test with pytest

on: 
  push:
    branches:
      - main
      - dev
      - sd3
  pull_request:
    branches:
      - main
      - dev
      - sd3

# CKV2_GHA_1: "Ensure top-level permissions are not set to write-all"
permissions: read-all

jobs:
  build:
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ubuntu-latest]
        python-version: ["3.10"] # Python versions to test
        pytorch-version: ["2.6.0", "2.8.0"] # PyTorch versions to test (diffusers >= 0.40 requires torch >= 2.6)

    steps:
      - uses: actions/checkout@v4
        with:
          # https://woodruffw.github.io/zizmor/audits/#artipacked
          persist-credentials: false

      - uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}
          cache: 'pip' 

      - name: Install and update pip, setuptools, wheel
        run: |
          # Setuptools, wheel for compiling some packages
          python -m pip install --upgrade pip setuptools wheel

      - name: Install dependencies
        run: |
          # Pre-install torch to pin version (requirements.txt has dependencies like transformers which requires pytorch)
          pip install dadaptation==3.2 torch==${{ matrix.pytorch-version }} torchvision pytest==8.3.4
          pip install -r requirements.txt

      - name: Check that the pinned PyTorch version is still installed
        run: |
          # If a requirement needs a newer torch, pip silently upgrades torch and leaves torchvision behind,
          # which breaks every import ("operator torchvision::nms does not exist"). Fail loudly instead.
          python -c "import torch, torchvision; print(torch.__version__, torchvision.__version__); assert torch.__version__.startswith('${{ matrix.pytorch-version }}'), 'torch was upgraded by requirements.txt'"

      - name: Test with pytest
        run: pytest # See pytest.ini for configuration