colossalai

1
name: Publish Nightly Version to PyPI
2

3
on:
4
  workflow_dispatch:
5
  schedule:
6
    - cron:  '0 0 * * 6' # release on every Sunday 00:00 UTC time
7

8
jobs:
9
  publish:
10
    if: github.repository == 'hpcaitech/ColossalAI'
11
    name: Build and publish Python 🐍 distributions 📦 to PyPI
12
    runs-on: ubuntu-latest
13
    timeout-minutes: 20
14
    outputs:
15
      status: ${{ steps.publish.outcome }}
16
    steps:
17
    - uses: actions/checkout@v2
18

19
    - uses: actions/setup-python@v2
20
      with:
21
        python-version: '3.8.14'
22

23
    - run: |
24
        python .github/workflows/scripts/update_setup_for_nightly.py
25
        python setup.py sdist build
26

27
    # publish to PyPI if executed on the main branch
28
    - name: Publish package to PyPI
29
      uses: pypa/gh-action-pypi-publish@release/v1
30
      id: publish
31
      with:
32
        user: __token__
33
        password: ${{ secrets.PYPI_API_TOKEN }}
34
        verbose: true
35

36
  notify:
37
    name: Notify Lark via webhook
38
    needs: publish
39
    runs-on: ubuntu-latest
40
    if: ${{ always() }} && github.repository == 'hpcaitech/ColossalAI'
41
    steps:
42
      - uses: actions/checkout@v2
43

44
      - uses: actions/setup-python@v2
45
        with:
46
          python-version: '3.8.14'
47

48
      - name: Install requests
49
        run: pip install requests
50

51
      - name: Notify Lark
52
        id: message-preparation
53
        run: |
54
          url=$SERVER_URL/$REPO/actions/runs/$RUN_ID
55

56
          if [ $STATUS == 'success' ]
57
          then
58
            msg="The Colossal-AI nightly version has been successfully released to PyPI."
59
          else
60
            msg="Failed to release Colossal-AI nightly version to PyPI, please visit $url for details."
61
          fi
62
          echo $msg
63
          python .github/workflows/scripts/send_message_to_lark.py -m "$msg" -u $WEBHOOK_URL
64
        env:
65
          SERVER_URL: ${{github.server_url }}
66
          REPO: ${{ github.repository }}
67
          RUN_ID: ${{ github.run_id }}
68
          WEBHOOK_URL: ${{ secrets.LARK_NOTIFICATION_WEBHOOK_URL }}
69
          STATUS: ${{ needs.publish.outputs.status }}
70