feat(ops): add accessor APIs tree_flatten_with_accessor and `PyTree…
#1003
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| name: Build | |
| on: | |
| push: | |
| branches: | |
| - main # allow to trigger the workflow with tag push event | |
| pull_request: | |
| paths: | |
| - setup.py | |
| - setup.cfg | |
| - pyproject.toml | |
| - MANIFEST.in | |
| - CMakeLists.txt | |
| - include/** | |
| - src/** | |
| - optree/version.py | |
| - .github/workflows/build.yml | |
| release: | |
| types: | |
| - published | |
| # Allow to trigger the workflow manually | |
| workflow_dispatch: | |
| inputs: | |
| task: | |
| description: "Task type" | |
| type: choice | |
| options: | |
| - build-only | |
| - build-and-publish | |
| required: true | |
| permissions: | |
| contents: read | |
| concurrency: | |
| group: "${{ github.workflow }}-${{ github.ref }}" | |
| cancel-in-progress: ${{ github.event_name == 'pull_request' }} | |
| jobs: | |
| build-sdist: | |
| name: Build sdist | |
| runs-on: ubuntu-latest | |
| if: github.repository == 'metaopt/optree' && (github.event_name != 'push' || startsWith(github.ref, 'refs/tags/')) | |
| timeout-minutes: 10 | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: "recursive" | |
| fetch-depth: 0 | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: "3.7 - 3.12" | |
| update-environment: true | |
| - name: Upgrade pip | |
| run: | | |
| python -m pip install --upgrade pip setuptools wheel pybind11 | |
| - name: Set __release__ | |
| if: | | |
| startsWith(github.ref, 'refs/tags/') || | |
| (github.event_name == 'workflow_dispatch' && github.event.inputs.task == 'build-and-publish') | |
| run: | | |
| python .github/workflows/set_release.py | |
| - name: Print version | |
| run: python setup.py --version | |
| - name: Install dependencies | |
| run: python -m pip install --upgrade pip setuptools wheel build | |
| - name: Build sdist | |
| run: python -m build --sdist | |
| - name: Upload artifact | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: sdist | |
| path: dist/*.tar.gz | |
| if-no-files-found: error | |
| build-wheels-py37: | |
| name: Build wheels for Python ${{ matrix.python-version }} on ${{ matrix.os }} | |
| runs-on: ${{ matrix.os }} | |
| needs: [build-sdist] | |
| if: github.repository == 'metaopt/optree' && (github.event_name != 'push' || startsWith(github.ref, 'refs/tags/')) | |
| strategy: | |
| matrix: | |
| os: [ubuntu-latest, windows-latest, macos-latest] | |
| python-version: ["3.7"] | |
| include: | |
| - os: macos-13 | |
| python-version: "3.7" | |
| exclude: | |
| - os: macos-latest | |
| python-version: "3.7" # Python 3.7 does not support macOS ARM64 | |
| fail-fast: false | |
| timeout-minutes: 45 | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: "recursive" | |
| fetch-depth: 0 | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| update-environment: true | |
| - name: Upgrade pip | |
| run: | | |
| python -m pip install --upgrade pip setuptools wheel pybind11 | |
| - name: Set __release__ | |
| if: | | |
| startsWith(github.ref, 'refs/tags/') || | |
| (github.event_name == 'workflow_dispatch' && github.event.inputs.task == 'build-and-publish') | |
| run: python .github/workflows/set_release.py | |
| - name: Print version | |
| run: python setup.py --version | |
| - name: Set CIBW_BUILD | |
| run: python .github/workflows/set_cibw_build.py | |
| - name: Set up QEMU | |
| if: runner.os == 'Linux' | |
| uses: docker/setup-qemu-action@v3 | |
| with: | |
| platforms: all | |
| - name: Build wheels | |
| uses: pypa/cibuildwheel@v2.17 | |
| env: | |
| CIBW_BUILD: ${{ env.CIBW_BUILD }} | |
| CIBW_ARCHS_LINUX: auto64 aarch64 | |
| CIBW_ARCHS_WINDOWS: auto64 ARM64 | |
| CIBW_ARCHS_MACOS: x86_64 arm64 universal2 | |
| with: | |
| package-dir: . | |
| output-dir: wheelhouse | |
| config-file: "{package}/pyproject.toml" | |
| - uses: actions/upload-artifact@v4 | |
| with: | |
| name: wheels-${{ matrix.python-version }}-${{ matrix.os }} | |
| path: wheelhouse/*.whl | |
| if-no-files-found: error | |
| build-wheels: | |
| name: Build wheels for Python ${{ matrix.python-version }} on ${{ matrix.os }} | |
| runs-on: ${{ matrix.os }} | |
| needs: [build-sdist, build-wheels-py37] | |
| if: github.repository == 'metaopt/optree' && (github.event_name != 'push' || startsWith(github.ref, 'refs/tags/')) | |
| strategy: | |
| matrix: | |
| os: [ubuntu-latest, windows-latest, macos-latest] | |
| python-version: ["3.8", "3.9", "3.10", "3.11", "3.12"] | |
| fail-fast: false | |
| timeout-minutes: 45 | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: "recursive" | |
| fetch-depth: 0 | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| with: | |
| python-version: ${{ matrix.python-version }} | |
| update-environment: true | |
| - name: Upgrade pip | |
| run: | | |
| python -m pip install --upgrade pip setuptools wheel pybind11 | |
| - name: Set __release__ | |
| if: | | |
| startsWith(github.ref, 'refs/tags/') || | |
| (github.event_name == 'workflow_dispatch' && github.event.inputs.task == 'build-and-publish') | |
| run: python .github/workflows/set_release.py | |
| - name: Print version | |
| run: python setup.py --version | |
| - name: Set CIBW_BUILD | |
| run: python .github/workflows/set_cibw_build.py | |
| - name: Set up QEMU | |
| if: runner.os == 'Linux' | |
| uses: docker/setup-qemu-action@v3 | |
| with: | |
| platforms: all | |
| - name: Build wheels | |
| uses: pypa/cibuildwheel@v2.17 | |
| env: | |
| CIBW_BUILD: ${{ env.CIBW_BUILD }} | |
| CIBW_ARCHS_LINUX: auto64 aarch64 | |
| CIBW_ARCHS_WINDOWS: auto64 ARM64 | |
| CIBW_ARCHS_MACOS: x86_64 arm64 universal2 | |
| with: | |
| package-dir: . | |
| output-dir: wheelhouse | |
| config-file: "{package}/pyproject.toml" | |
| - uses: actions/upload-artifact@v4 | |
| with: | |
| name: wheels-${{ matrix.python-version }}-${{ matrix.os }} | |
| path: wheelhouse/*.whl | |
| if-no-files-found: error | |
| list-artifacts: | |
| name: List artifacts | |
| runs-on: ubuntu-latest | |
| needs: [build-sdist, build-wheels-py37, build-wheels] | |
| if: github.repository == 'metaopt/optree' && (github.event_name != 'push' || startsWith(github.ref, 'refs/tags/')) | |
| timeout-minutes: 15 | |
| steps: | |
| - name: Download built sdist | |
| uses: actions/download-artifact@v4 | |
| with: | |
| # unpacks default artifact into dist/ | |
| # if `name: artifact` is omitted, the action will create extra parent dir | |
| name: sdist | |
| path: dist | |
| - name: Download built wheels | |
| uses: actions/download-artifact@v4 | |
| with: | |
| pattern: wheels-* | |
| path: dist | |
| merge-multiple: true | |
| - name: List distributions | |
| run: ls -lh dist/* | |
| - name: Upload artifacts | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: artifacts | |
| path: dist/* | |
| if-no-files-found: error | |
| publish: | |
| runs-on: ubuntu-latest | |
| needs: [list-artifacts] | |
| if: | | |
| github.repository == 'metaopt/optree' && github.event_name != 'pull_request' && | |
| (github.event_name != 'workflow_dispatch' || github.event.inputs.task == 'build-and-publish') && | |
| (github.event_name != 'push' || startsWith(github.ref, 'refs/tags/')) | |
| timeout-minutes: 15 | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| with: | |
| submodules: "recursive" | |
| fetch-depth: 0 | |
| - name: Set up Python | |
| uses: actions/setup-python@v5 | |
| if: startsWith(github.ref, 'refs/tags/') | |
| with: | |
| python-version: "3.7 - 3.12" | |
| update-environment: true | |
| - name: Upgrade pip | |
| run: | | |
| python -m pip install --upgrade pip setuptools wheel pybind11 | |
| - name: Set __release__ | |
| if: startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' | |
| run: | | |
| python .github/workflows/set_release.py | |
| - name: Print version | |
| run: python setup.py --version | |
| - name: Check consistency between the package version and release tag | |
| if: startsWith(github.ref, 'refs/tags/') | |
| run: | | |
| PACKAGE_VER="v$(python setup.py --version)" | |
| RELEASE_TAG="${GITHUB_REF#refs/*/}" | |
| if [[ "${PACKAGE_VER}" != "${RELEASE_TAG}" ]]; then | |
| echo "package ver. (${PACKAGE_VER}) != release tag. (${RELEASE_TAG})" | |
| exit 1 | |
| fi | |
| - name: Download built artifacts | |
| uses: actions/download-artifact@v4 | |
| with: | |
| # unpacks default artifact into dist/ | |
| # if `name: artifact` is omitted, the action will create extra parent dir | |
| name: artifacts | |
| path: dist | |
| - name: List distributions | |
| run: ls -lh dist/* | |
| - name: Publish to TestPyPI | |
| if: startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' | |
| uses: pypa/gh-action-pypi-publish@release/v1 | |
| with: | |
| user: __token__ | |
| password: ${{ secrets.TESTPYPI_UPLOAD_TOKEN }} | |
| repository-url: https://test.pypi.org/legacy/ | |
| verbose: true | |
| print-hash: true | |
| skip-existing: true | |
| - name: Publish to PyPI | |
| if: startsWith(github.ref, 'refs/tags/') || github.event_name == 'workflow_dispatch' | |
| uses: pypa/gh-action-pypi-publish@release/v1 | |
| with: | |
| user: __token__ | |
| password: ${{ secrets.PYPI_UPLOAD_TOKEN }} | |
| verbose: true | |
| print-hash: true | |
| skip-existing: true |