python-publish-gpu.yml

name: Build and publish GPU 🐍📦 to PyPI

on:
  push:
    tags:
    - 'v*'

jobs:

  test:
    name: Test before publish
    uses: ./.github/workflows/tests.yml

  build-n-publish-gpu:
    needs: test
    if: startsWith(github.ref, 'refs/tags/')
    name: Build and publish GPU 🐍📦 to PyPI
    runs-on: ubuntu-latest
    environment:
      name: pypi-gpu
      url: https://pypi.org/p/anylabeling-gpu
    permissions:
      id-token: write  # IMPORTANT: this permission is mandatory for trusted publishing
    steps:
    - uses: actions/checkout@v4
    - name: Set up Python
      uses: actions/setup-python@v5
      with:
        python-version: "3.x"
    - name: Install pypa/build
      run: >-
        python -m pip install build==1.2.2 twine==6.1.0 --user
    - name: Set preferred device to GPU
      run: >-
        sed -i'' -e 's/\_\_preferred_device\_\_[ ]*=[ ]*\"[A-Za-z0-9]*\"/__preferred_device__ = "GPU"/g' anylabeling/app_info.py

    # PEP 621 makes pyproject.toml `[project]` metadata authoritative — setup.py
    # cannot override `name` or replace `dependencies`. Rewrite pyproject.toml
    # in place so the GPU wheel ships as `anylabeling-gpu` with onnxruntime-gpu.
    - name: Rewrite pyproject.toml for the GPU package
      run: |
        sed -i 's/^name = "anylabeling"$/name = "anylabeling-gpu"/' pyproject.toml
        sed -i 's/"onnxruntime>=1.20.0"/"onnxruntime-gpu>=1.20.0"/' pyproject.toml
        echo "--- after rewrite ---"
        grep -E '^name|onnxruntime' pyproject.toml

    - name: Build a binary wheel and a source tarball
      run: >-
        python -m build --wheel --outdir dist/ .

    - name: Verify built wheel is anylabeling-gpu
      run: |
        ls dist/
        whl=$(ls dist/anylabeling_gpu-*.whl)
        python -m zipfile -e "$whl" /tmp/whl_extract/
        grep -E '^Name:|^Requires-Dist: onnxruntime' /tmp/whl_extract/anylabeling_gpu-*.dist-info/METADATA
    - name: Publish distribution 📦 to PyPI
      if: startsWith(github.ref, 'refs/tags')
      uses: pypa/gh-action-pypi-publish@release/v1
      with:
        skip-existing: true