Skip to content

fix(data-gen): build only the tpchgen-cli Rust stage - #402

Open
Avinash-Raj wants to merge 1 commit into
mainfrom
fix/tpchgen-cli-skip-pyarrow-stage
Open

fix(data-gen): build only the tpchgen-cli Rust stage#402
Avinash-Raj wants to merge 1 commit into
mainfrom
fix/tpchgen-cli-skip-pyarrow-stage

Conversation

@Avinash-Raj

Copy link
Copy Markdown
Contributor

Summary

install_tpchgen_cli.sh clones TomAugspurger/tpchgen-rs and docker builds the upstream Dockerfile, then copies only /usr/local/bin/tpchgen-cli out of the image.

That Dockerfile is multi-stage. The builder stage compiles the Rust CLI. A later python-deps stage runs uv pip install pyarrow (~48 MiB wheel) with UV_HTTP_TIMEOUT=30s. CI often cannot finish that download/extract, so the whole install fails even though we never use pyarrow or the Python entrypoint.

This change builds --target builder so Cargo still produces tpchgen-cli and the pyarrow install is skipped.

Why?

Our CI runner job (https://github.com/rapidsai/velox-testing/actions/runs/32334525240/job/96331784081) sometimes faces network failures when building the unnecessary python-deps stage. This PR restricts Docker to build only the necessary stage (builder) instead of all stages such as builder, python-deps, etc., since we only need the binary built from the builder stage.

@Avinash-Raj
Avinash-Raj requested a review from a team as a code owner August 20, 2026 16:28
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant