Fix auto-publish tool, replace setup.py with pyproject.toml standard,…

… disable duplicate builds.

PiperOrigin-RevId: 490342029

.github/workflows/auto-publish.yml

@@ -3,43 +3,13 @@ name: Auto-publish
 on: [push, workflow_dispatch]
 
 jobs:
-  pytest-job:
-    runs-on: ubuntu-latest
-    strategy:
-      matrix:
-        # LINT.IfChange
-        python-version: [3.7, 3.8, 3.9]
-        # LINT.ThenChange(build.yml)
-
-    timeout-minutes: 30
-
-    steps:
-    - uses: styfle/[email protected]
-      with:
-        access_token: ${{ github.token }}
-
-    - uses: actions/checkout@v3
-
-    # Install deps
-    - uses: actions/setup-python@v2
-      with:
-        python-version: ${{ matrix.python-version }}
-    - run: pip --version
-    - run: pip install -e .[all,dev]
-    - run: pip freeze
-
-    # Run tests
-    - name: Run core tests
-      run: pytest -vv -n auto
-
   publish-job:
     # Only try to publish if:
     # * Repo is self (prevents running from forks)
     # * Branch is `main`
     if: |
       github.repository == 'google/orbax'
       && github.ref == 'refs/heads/main'
-    needs: pytest-job  # Only publish after tests are successful
     runs-on: ubuntu-latest
     permissions:
       contents: write

.github/workflows/python-publish.yml

@@ -23,9 +23,6 @@ jobs:
         python -m pip install --upgrade pip
         pip install setuptools wheel twine
     - name: Build and publish
-      env:
-        TWINE_USERNAME: __token__
-        TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
-      run: |
-        python setup.py sdist bdist_wheel
-        twine upload dist/*
+      uses: etils-actions/pypi-build-publish@v1
+      with:
+        pypi-token: ${{ secrets.PYPI_API_TOKEN }}

orbax/__init__.py

@@ -12,4 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-# Place holder for orbax package, does nothing.
+"""Orbax API."""
+
+# A new PyPI release will be pushed everytime `__version__` is increased.
+__version__ = '0.0.18'

pyproject.toml

@@ -0,0 +1,47 @@
+[build-system]
+requires = ["flit_core >=3.5,<4"]
+build-backend = "flit_core.buildapi"
+
+[project]
+name = 'orbax'
+description='Orbax'
+readme = 'README.md'
+license = {file = 'LICENSE'}
+requires-python = '>=3.7'
+authors = [{name = 'Orbax Authors', email='[email protected]'}]
+classifiers=[
+    'Development Status :: 4 - Beta',
+    'Intended Audience :: Developers',
+    'Intended Audience :: Science/Research',
+    'License :: OSI Approved :: Apache Software License',
+    'Programming Language :: Python',
+    'Topic :: Scientific/Engineering :: Artificial Intelligence',
+]
+
+keywords = ['JAX machine learning']
+
+dependencies = [
+    'absl-py',
+    'cached_property',
+    'importlib_resources',
+    'etils',
+    'flax',
+    'importlib_resources',
+    'jax',
+    'jaxlib',
+    'numpy',
+    'pytest',
+    'pyyaml',
+    'tensorstore >= 0.1.20',
+]
+
+dynamic = ['version']
+
+[project.urls]
+homepage = 'http://github.com/google/orbax'
+repository = 'http://github.com/google/orbax'
+
+[project.optional-dependencies]
+dev = [
+    'pytest-xdist'
+]