test-infra/.github/workflows/release-pypi.yml at c423bd5ba73727dd6d6cf910835c2359324cc954 · pytorch/test-infra · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
name: Release binary to pypi

on:
  workflow_dispatch:
    inputs:
      dryrun:
        required: true
        type: choice
        default: enabled
        options:
          - enabled
          - disabled
      package:
        description: "Domain to prepare and release"
        required: true
        type: choice
        default: torchao
        options:
          - torchao
          - torchtune
          - executorch
          - torchvision
          - torchaudio
          - torchcodec

permissions:
  id-token: write
  contents: read

jobs:
  trigger:
    runs-on: ubuntu-latest
    timeout-minutes: 60
    environment: pytorchbot-env
    steps:
      - uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
      - name: Configure aws credentials (pytorch account)
        uses: aws-actions/configure-aws-credentials@50ac8dd1e1b10d09dac7b8727528b91bed831ac0 # v3.0.2
        with:
          role-to-assume: arn:aws:iam::749337293305:role/gha_workflow_stage_wheels
          aws-region: us-east-1
      - name: Install AWS CLI v2
        run: |
          curl "https://awscli.amazonaws.com/awscli-exe-linux-x86_64.zip" -o /tmp/awscliv2.zip
          unzip -q /tmp/awscliv2.zip -d /tmp
          sudo /tmp/aws/install --update
          rm -rf /tmp/aws/ /tmp/awscliv2.zip
      - name: Verify AWS CLI installation
        run: aws --version
      - name: Copy staged binaries
        shell: bash
        env:
          PACKAGE: ${{ inputs.package || 'torchvision' }}
          DRY_RUN: ${{ inputs.dryrun || 'enabled' }}
        run: |
            set -ex

            # Init release versions variables
            source ./release/release_versions.sh
            # shellcheck disable=SC2086
            version="${PACKAGE^^}_VERSION"
            mkdir dist/
            # shellcheck disable=SC2086
            aws s3 sync "s3://pytorch-backup/${PACKAGE}-${!version}-pypi-staging/" dist/ --debug
      - name: Display structure of downloaded files
        run: ls -R dist/
      - name: Publish package to PyPI
        if: ${{ inputs.dryrun == 'disabled' }}
        uses: pypa/gh-action-pypi-publish@release/v1
        with:
          verbose: true