Create a dev pypi push variant.

PiperOrigin-RevId: 623978390

Author: xingyousong

.github/workflows/pypi-publish-dev.yml

@@ -0,0 +1,33 @@
+# This workflow will upload a dev Python Package to TestPyPI using Twine when a main branch commit
+# is pushed. For more information see: https://help.github.com/en/actions/language-and-framework-guides/using-python-with-github-actions#publishing-to-package-registries
+# NOTE: User will need to add a extra url to `requirements.txt`: https://stackoverflow.com/questions/29281587/format-a-requirements-txt-file-for-pip-where-one-or-more-packages-have-a-differe
+
+name: pypi-dev
+
+# PyPI push triggers whenever we push a commit to main.
+on:
+  push:
+    branches:
+      - main
+
+jobs:
+  deploy:
+    runs-on: ubuntu-latest
+    steps:
+    - uses: actions/checkout@v3
+    - name: Set up Python
+      uses: actions/setup-python@v4
+      with:
+        python-version: '3.10'  # 3.11 breaks grpcio-tools.
+    - name: Install dependencies
+      run: |
+        sudo apt-get install -y libprotobuf-dev
+        python -m pip install --upgrade pip
+        pip install setuptools wheel twine grpcio-tools==1.48.2
+    - name: Build and publish
+      env:
+        TWINE_USERNAME: ${{ secrets.PYPI_USERNAME }}
+        TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
+      run: |
+        python setup.py --dev sdist bdist_wheel
+        twine upload --repository testpypi dist/*

setup.py

@@ -15,6 +15,8 @@
 from __future__ import annotations
 
 """Setup for pip package."""
+
+import datetime
 import itertools
 import os
 import sys
@@ -68,6 +70,12 @@ def run(self):
 
 
 _VERSION = _get_version()
+_NAME = 'google-vizier'
+
+if '--dev' in sys.argv:
+  sys.argv.remove('--dev')
+  _VERSION += '.dev' + datetime.datetime.now().strftime('%Y%m%d-%H%M%S')
+  _NAME += '-dev'
 
 extras_require = {
     'jax': _parse_requirements('requirements-jax.txt'),
@@ -82,7 +90,7 @@ def run(self):
 )
 
 setup(
-    name='google-vizier',
+    name=_NAME,
     version=_VERSION,
     url='https://github.com/google/vizier',
     license='Apache License 2.0',