Upload wheels to PyPI from GitHub workflows

.github/CHANGELOG.md

@@ -6,6 +6,8 @@
 
 ### Improvements
 
+* Set GitHub workflow to upload wheels to Test PyPI [(#220)](https://github.com/PennyLaneAI/pennylane-lightning/pull/220)
+
 ### Documentation
 
 ### Bug fixes
@@ -14,6 +16,8 @@
 
 This release contains contributions from (in alphabetical order):
 
+Chae-Yeun Park
+
 ---
 
 # Release 0.21.0

.github/workflows/dev_version_script.py

@@ -0,0 +1,80 @@
+# Copyright 2022 Xanadu Quantum Technologies Inc.
+
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+
+#     http://www.apache.org/licenses/LICENSE-2.0
+
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+import argparse
+from pathlib import Path
+import importlib
+
+import re
+
+VERSION_FILE_PATH = 'pennylane_lightning/_version.py'
+
+rgx_ver = re.compile('^__version__ = \"(.*?)\"$')
+
+rgx_dev_ver = re.compile('^(\d*\.\d*\.\d*)-dev(\d*)$')
+
+def extract_version(package_path):
+    with package_path.joinpath(VERSION_FILE_PATH).open('r') as f:
+        for line in f.readlines():
+            if line.startswith('__version__'):
+                line = line.strip()
+                m = rgx_ver.match(line)
+                return m.group(1)
+    raise ValueError("Cannot parse version")
+
+def is_dev(version_str):
+    m = rgx_dev_ver.fullmatch(version_str)
+    if m:
+        return True
+    else:
+        return False
+
+def update_dev_version(package_path, version_str):
+    m = rgx_dev_ver.fullmatch(version_str)
+    if m.group(2) == '':
+        curr_dev_ver = 0
+    else:
+        curr_dev_ver = int(m.group(2))
+
+    new_version_str = '{}-dev{}'.format(m.group(1), str(curr_dev_ver + 1))
+
+    lines = []
+    with package_path.joinpath(VERSION_FILE_PATH).open('r') as f:
+        for line in f.readlines():
+            if not line.startswith('__version__'):
+                lines.append(line)
+            else:
+                lines.append(f'__version__ = \"{new_version_str}\"\n')
+
+    with package_path.joinpath(VERSION_FILE_PATH).open('w') as f:
+        f.write(''.join(lines))
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser()
+    parser.add_argument("--pr-path", dest = "pr", type=str, required=True, help="Path to the PR dir")
+    parser.add_argument("--master-path", dest = "master", type=str, required=True, help="Path to the master dir")
+
+    args = parser.parse_args()
+
+    pr_version = extract_version(Path(args.pr))
+    master_version = extract_version(Path(args.master))
+
+    if pr_version == master_version:
+        if is_dev(pr_version):
+            print("Automatically update version string.")
+            update_dev_version(Path(args.pr), pr_version)
+        else:
+            print("Even though version of this PR is different from the master, as the PR is not dev, we do nothing.")
+    else:
+        print("Version of this PR is already different from master. Do nothing.")

.github/workflows/update_dev_version.yml

@@ -0,0 +1,44 @@
+name: Update dev version automatically
+on:
+  pull_request:
+
+jobs:
+  update-dev-version:
+    name: Update version for development branches
+    runs-on: ubuntu-latest
+    steps:
+
+      - name: Checkout PennyLane-Lightning master
+        uses: actions/checkout@v2
+        with:
+          ref: master
+          path: main
+
+      - uses: actions/setup-python@v2
+        name: Install Python
+        with:
+          python-version: '3.7'
+
+      - name: Checkout PennyLane-Lightning PR
+        uses: actions/checkout@v2
+        with:
+          token: ${{ secrets.DEV_TOKEN }}
+          ref: ${{ github.head_ref }}
+          path: pr
+
+      - name: Run version update script
+        run: >
+          python3 pr/.github/workflows/dev_version_script.py
+          --pr-path "./pr" --master-path "./main"
+
+      - name: Commit and push changes
+        if:
+        run: |
+          cd ./pr
+          if [[ -n $(git status -s) ]]; then
+            git config --local user.email "chae-yeun@xanadu.ai"
+            git config --local user.name "Dev version update bot"
+            git add .
+            git commit -m 'Auto update version'
+            git push
+          fi

.github/workflows/wheel_linux_aarch64.yml

@@ -62,3 +62,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: linux-wheels-aarch64
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: Linux-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_linux_ppc64le.yml

@@ -59,3 +59,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: linux-wheels-ppc64le
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: Linux-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_linux_x86_64.yml

@@ -64,3 +64,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: linux-wheels-x86-64
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: Linux-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_macos_arm64.yml

@@ -62,3 +62,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: mac-wheels-arm64
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: macOS-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_macos_x86_64.yml

@@ -71,3 +71,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: mac-wheels-x86
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master' }}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: macOS-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_noarch.yml

@@ -38,5 +38,22 @@ jobs:
       - uses: actions/upload-artifact@v2
         if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master' }}
         with:
-          name: pure-python-wheels
+          name: pure-python-wheels.zip
           path: main/dist/*.whl
+
+  upload-pypi:
+    needs: build-pure-python-wheel
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: pure-python-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

.github/workflows/wheel_win_x86_64.yml

@@ -73,3 +73,20 @@ jobs:
         with:
           name: ${{ runner.os }}-wheels.zip
           path: ./wheelhouse/*.whl
+
+  upload-pypi:
+    needs: win-wheels
+    runs-on: ubuntu-latest
+    if: ${{ github.event_name == 'release' || github.ref == 'refs/heads/master'}}
+    steps:
+      - uses: actions/download-artifact@v2
+        with:
+          name: Windows-wheels.zip
+          path: dist
+
+      - name: Upload wheels to PyPI
+        uses: pypa/gh-action-pypi-publish@release/v1
+        with:
+          user: __token__
+          password: ${{ secrets.TEST_PYPI_API_TOKEN }}
+          repository_url: https://test.pypi.org/legacy/

pennylane_lightning/_version.py

@@ -16,4 +16,4 @@
    Version number (major.minor.patch[-label])
 """
 
-__version__ = "0.22.0-dev"
+__version__ = "0.22.0-dev1"