fix: Correct Docker build for rustbpe tokenizer

This commit fixes a build failure in the Docker image by adding the `--uv` flag to the `maturin develop` command.

The `maturin` build process was failing because it could not find `pip` within the `uv` environment. The `--uv` flag ensures that `maturin` correctly uses the `uv` environment to build the `rustbpe` tokenizer.
This commit is contained in:
google-labs-jules[bot] 2025-11-04 02:05:34 +00:00
parent a2189d20d0
commit fa04262889

View File

@ -17,7 +17,7 @@ COPY . .
RUN /root/.local/bin/uv sync --extra gpu
# Build the rustbpe tokenizer.
RUN /root/.local/bin/uv run maturin develop --release --manifest-path rustbpe/Cargo.toml
RUN /root/.local/bin/uv run maturin develop --release --manifest-path rustbpe/Cargo.toml --uv
# Set the entrypoint.
ENTRYPOINT ["python"]