name: Test with pytest

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

jobs:
  build:
    runs-on: ${{ matrix.os }}
    strategy:
      matrix:
        os: [ubuntu-latest]
        python-version: ["3.10"] # Python versions to test
        pytorch-version: ["2.4.0"] # PyTorch versions to test

    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==0.19.0 pytest==8.3.4
          pip install -r requirements.txt

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