Prevent panic when retry attempt value in map task gets larger than bitarray was allocated for #7159
This file contains hidden or 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 & Push Flyte Single Binary Images | |
| concurrency: | |
| group: ${{ github.workflow }}-${{ github.event.pull_request.number }} | |
| cancel-in-progress: true | |
| on: | |
| pull_request: | |
| push: | |
| branches: | |
| - master | |
| - rc/* | |
| - "release-v**" | |
| workflow_dispatch: | |
| jobs: | |
| test-bootstrap: | |
| runs-on: ubuntu-latest | |
| steps: | |
| - uses: actions/checkout@v4 | |
| - uses: actions/setup-go@v5 | |
| with: | |
| go-version-file: go.mod | |
| - name: golangci-lint | |
| uses: golangci/golangci-lint-action@v3 | |
| with: | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| - name: Check formatting | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| run: | | |
| make check-fmt | |
| - name: Test | |
| working-directory: docker/sandbox-bundled/bootstrap | |
| run: | | |
| make test | |
| build-and-push-single-binary-image: | |
| runs-on: ubuntu-latest | |
| needs: [test-bootstrap] | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - name: Setup Golang caches | |
| uses: actions/cache@v4 | |
| with: | |
| path: | | |
| /root/.cache/go-build | |
| /root/go/pkg/mod | |
| key: ${{ runner.os }}-golang-${{ hashFiles('go.sum') }} | |
| restore-keys: | | |
| ${{ runner.os }}-golang- | |
| - name: Set versions | |
| id: set_version | |
| run: | | |
| # TODO: Find a better way to determine the flyteconsole version | |
| echo "FLYTECONSOLE_VERSION=$(yq eval '.flyteconsole.image.tag' charts/flyte-core/values.yaml)" >> $GITHUB_ENV | |
| echo "FLYTE_VERSION=${{ github.sha }}" >> $GITHUB_ENV | |
| - name: Prepare Image Names | |
| id: image-names | |
| uses: docker/metadata-action@v3 | |
| with: | |
| # list of Docker images to use as base name for tags | |
| images: | | |
| ghcr.io/${{ github.repository_owner }}/flyte-binary | |
| tags: | | |
| type=raw,value=latest,enable=${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }} | |
| type=sha,format=long | |
| - name: Set up QEMU | |
| uses: docker/setup-qemu-action@v2 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| - name: Setup destination directories for image tarballs | |
| run: | | |
| mkdir -p docker/sandbox-bundled/images/tar/{arm64,amd64} | |
| - name: Export ARM64 Image | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| platforms: linux/arm64 | |
| tags: flyte-binary:sandbox | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| outputs: type=docker,dest=docker/sandbox-bundled/images/tar/arm64/flyte-binary.tar | |
| - name: Export AMD64 Image | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| platforms: linux/amd64 | |
| tags: flyte-binary:sandbox | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| outputs: type=docker,dest=docker/sandbox-bundled/images/tar/amd64/flyte-binary.tar | |
| - name: Upload single binary image | |
| uses: actions/upload-artifact@v4 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Login to GitHub Container Registry | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ghcr.io | |
| username: "${{ secrets.FLYTE_BOT_USERNAME }}" | |
| password: "${{ secrets.FLYTE_BOT_PAT }}" | |
| - name: Build and push Image | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: . | |
| platforms: linux/arm64, linux/amd64 | |
| tags: ${{ steps.image-names.outputs.tags }} | |
| build-args: | | |
| FLYTECONSOLE_VERSION=${{ env.FLYTECONSOLE_VERSION }} | |
| FLYTE_VERSION=${{ env.FLYTE_VERSION }} | |
| file: Dockerfile | |
| push: true | |
| sandbox-bundled-functional-tests: | |
| runs-on: ubuntu-latest | |
| env: | |
| FLYTESNACKS_PRIORITIES: "P0" | |
| FLYTESNACKS_VERSION: "" | |
| timeout-minutes: 60 | |
| needs: [build-and-push-single-binary-image] | |
| steps: | |
| - name: Set latest Flytesnacks release | |
| if: ${{ env.FLYTESNACKS_VERSION == '' }} | |
| run: | | |
| FLYTESNACKS_VERSION="$(curl --silent https://api.github.com/repos/flyteorg/flytesnacks/releases/latest | jq -r .tag_name)" | |
| echo "FLYTESNACKS_VERSION=${FLYTESNACKS_VERSION}" >> ${GITHUB_ENV} | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - uses: actions/download-artifact@v4 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| driver-opts: image=moby/buildkit:master | |
| buildkitd-flags: "--allow-insecure-entitlement security.insecure" | |
| - name: Build sandbox image for functional tests | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: docker/sandbox-bundled | |
| load: true | |
| allow: "security.insecure" | |
| tags: flyte-sandbox-bundled:local | |
| # Without this, the GHA runner is under disk pressure and evicts all the pods. | |
| # Buildx cache uses roughly 50% (7gb) of the GHA runner's disk (14gb). | |
| - name: Prune Docker Buildx cache to reclaim storage | |
| run: docker buildx prune --all --force | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: "3.12" | |
| - uses: unionai/flytectl-setup-action@v0.0.3 | |
| - name: Setup sandbox | |
| run: | | |
| mkdir -p ~/.flyte/sandbox | |
| cat << EOF > ~/.flyte/sandbox/config.yaml | |
| task_resources: | |
| defaults: | |
| cpu: "0" | |
| memory: "0" | |
| limits: | |
| cpu: "0" | |
| memory: "0" | |
| EOF | |
| flytectl demo start --image flyte-sandbox-bundled:local --imagePullPolicy Never | |
| - name: Install Python dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install uv | |
| uv pip install --system flytekit==1.15.0 flytekitplugins-deck-standard==1.15.0 "numpy<2.0.0" pyarrow pandas | |
| uv pip freeze | |
| - name: Checkout flytesnacks | |
| uses: actions/checkout@v4 | |
| with: | |
| repository: flyteorg/flytesnacks | |
| path: flytesnacks | |
| ref: ${{ env.FLYTESNACKS_VERSION }} | |
| - name: Register specific tests | |
| run: | | |
| flytekit_version=$(pip show flytekit | grep -i version | awk '{ print $2 }') | |
| while read -r line; | |
| do | |
| pyflyte -vv --config ./boilerplate/flyte/end2end/functional-test-config.yaml \ | |
| register \ | |
| --project flytesnacks \ | |
| --domain development \ | |
| --image cr.flyte.org/flyteorg/flytekit:py3.12-${flytekit_version} \ | |
| --version ${{ env.FLYTESNACKS_VERSION }} \ | |
| flytesnacks/$line; | |
| done < flytesnacks/flyte_tests.txt | |
| - name: Install Pytest | |
| run: | | |
| pip install pytest | |
| - name: End2End | |
| run: | | |
| make end2end_execute | |
| sandbox-bundled-offloaded-functional-tests: | |
| runs-on: ubuntu-latest | |
| env: | |
| FLYTESNACKS_PRIORITIES: "P0" | |
| FLYTESNACKS_VERSION: "" | |
| timeout-minutes: 60 | |
| needs: [build-and-push-single-binary-image] | |
| steps: | |
| - name: Set latest Flytesnacks release | |
| if: ${{ env.FLYTESNACKS_VERSION == '' }} | |
| run: | | |
| FLYTESNACKS_VERSION="$(curl --silent https://api.github.com/repos/flyteorg/flytesnacks/releases/latest | jq -r .tag_name)" | |
| echo "FLYTESNACKS_VERSION=${FLYTESNACKS_VERSION}" >> ${GITHUB_ENV} | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - uses: actions/download-artifact@v4 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| driver-opts: image=moby/buildkit:master | |
| buildkitd-flags: "--allow-insecure-entitlement security.insecure" | |
| - name: Build sandbox image for functional tests | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: docker/sandbox-bundled | |
| load: true | |
| allow: "security.insecure" | |
| tags: flyte-sandbox-bundled:local | |
| # Without this, the GHA runner is under disk pressure and evicts all the pods. | |
| # Buildx cache uses roughly 50% (7gb) of the GHA runner's disk (14gb). | |
| - name: Prune Docker Buildx cache to reclaim storage | |
| run: docker buildx prune --all --force | |
| - name: Set up Python | |
| uses: actions/setup-python@v4 | |
| with: | |
| python-version: "3.12" | |
| - uses: unionai/flytectl-setup-action@v0.0.3 | |
| - name: Setup sandbox | |
| run: | | |
| mkdir -p ~/.flyte/sandbox | |
| cat << EOF > ~/.flyte/sandbox/config.yaml | |
| propeller: | |
| literal-offloading-config: | |
| enabled: true | |
| min-size-in-mb-for-offloading: 0.000001 | |
| task_resources: | |
| defaults: | |
| cpu: "0" | |
| memory: "0" | |
| limits: | |
| cpu: "0" | |
| memory: "0" | |
| EOF | |
| flytectl demo start --image flyte-sandbox-bundled:local --imagePullPolicy Never | |
| # By setting min-size-in-mb-for-offloading to 0.000001, we ensure that all tasks are offloaded. | |
| - name: Install Python dependencies | |
| run: | | |
| python -m pip install --upgrade pip | |
| pip install uv | |
| uv pip install --system flytekit flytekitplugins-deck-standard "numpy<2.0.0" pyarrow pandas | |
| uv pip freeze | |
| - name: Checkout flytesnacks | |
| uses: actions/checkout@v4 | |
| with: | |
| repository: flyteorg/flytesnacks | |
| path: flytesnacks | |
| ref: ${{ env.FLYTESNACKS_VERSION }} | |
| - name: Register specific tests | |
| run: | | |
| flytekit_version=$(pip show flytekit | grep -i version | awk '{ print $2 }') | |
| while read -r line; | |
| do | |
| pyflyte -vv --config ./boilerplate/flyte/end2end/functional-test-config.yaml \ | |
| register \ | |
| --project flytesnacks \ | |
| --domain development \ | |
| --image cr.flyte.org/flyteorg/flytekit:py3.12-${flytekit_version} \ | |
| --version ${{ env.FLYTESNACKS_VERSION }} \ | |
| flytesnacks/$line; | |
| done < flytesnacks/flyte_tests.txt | |
| - name: Install Pytest | |
| run: | | |
| pip install pytest | |
| - name: End2End | |
| run: | | |
| make end2end_execute | |
| build-and-push-sandbox-bundled-image: | |
| if: ${{ github.event_name == 'push' || github.event_name == 'workflow_dispatch' }} | |
| runs-on: ubuntu-latest | |
| needs: [build-and-push-single-binary-image] | |
| steps: | |
| - name: Checkout | |
| uses: actions/checkout@v4 | |
| - uses: actions/download-artifact@v4 | |
| with: | |
| name: single-binary-image | |
| path: docker/sandbox-bundled/images/tar | |
| - name: Set up QEMU | |
| uses: docker/setup-qemu-action@v2 | |
| - name: Set up Docker Buildx | |
| uses: docker/setup-buildx-action@v2 | |
| with: | |
| driver-opts: image=moby/buildkit:master | |
| buildkitd-flags: "--allow-insecure-entitlement security.insecure" | |
| - name: Set version | |
| id: set_version | |
| run: | | |
| echo "FLYTE_SANDBOX_VERSION=${{ github.sha }}" >> $GITHUB_ENV | |
| - name: Prepare Image Names | |
| id: image-names | |
| uses: docker/metadata-action@v3 | |
| with: | |
| # list of Docker images to use as base name for tags | |
| images: | | |
| ghcr.io/${{ github.repository_owner }}/flyte-sandbox-bundled | |
| tags: | | |
| type=raw,value=latest,enable=${{ github.event_name == 'push' && github.ref == 'refs/heads/master' }} | |
| type=sha,format=long, | |
| - name: Login to GitHub Container Registry | |
| uses: docker/login-action@v2 | |
| with: | |
| registry: ghcr.io | |
| username: "${{ secrets.FLYTE_BOT_USERNAME }}" | |
| password: "${{ secrets.FLYTE_BOT_PAT }}" | |
| - name: Build and push multi-arch image | |
| uses: docker/build-push-action@v6 | |
| with: | |
| context: docker/sandbox-bundled | |
| allow: "security.insecure" | |
| platforms: linux/arm64, linux/amd64 | |
| tags: ${{ steps.image-names.outputs.tags }} | |
| build-args: "FLYTE_SANDBOX_VERSION=${{ env.FLYTE_SANDBOX_VERSION }}" | |
| push: true |