name: Build package

#
# This workflow is a test of the python package build.
# Install Python dependencies across different Python versions.
#

on: [ push ]

jobs:
  build:
    name: Build Test
    runs-on: ${{ matrix.runner.labels }}
    container: ${{ matrix.runner.container }}
    strategy:
      fail-fast: false
      matrix:
        runner:
          - labels: [self-hosted, Linux, X64, cpu]
            container: "ubuntu"
          - labels: [self-hosted, Linux, X64, ipex-a380-6gb]
            container: "intel/intel-optimized-pytorch:2.3.0-pip-base"
          - labels: [self-hosted, Linux, X64, rocm-7600-8gb]
            container: "rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2"
          - labels: [self-hosted, Linux, X64, cuda-3060-12gb]
            container: "nvcr.io/nvidia/pytorch:24.04-py3"
    steps:
      - run: |
          apt update
          apt install -y python3 python3-setuptools git build-essential ffmpeg libsm6 libxext6
          pip uninstall -y $(pip list --format=freeze | grep opencv) &&
              rm -rf /usr/local/lib/python3.10/dist-packages/cv2/ || rm -rf /usr/local/lib/python3.11/dist-packages/cv2/ || rm -rf /usr/local/lib/python3.12/dist-packages/cv2/
      - uses: actions/checkout@v4
      - name: Install dependencies
        run: |
          python -m pip install --upgrade pip
          pip install --no-build-isolation .[dev]
      - name: Run unit tests
        run: |
          export HSA_OVERRIDE_GFX_VERSION=11.0.0
          pytest -v tests/unit
      - name: Lint for errors
        run: |
          pylint comfy