maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Use venv and `uv` throughout in lieu of system Python env + `pip`

Open SamuelMarks opened this issue 1 month ago • 0 comments

Description

uv is inconsistently used in the codebase. In addition to being a faster complete rewrite in Rust, uv also has a dependency resolver that always works; whereas the usual pip one is sometimes flakey. And we don't test various Python and uv systems alongside each other anyway, so we might as well just use uv and later we can setup explicit tests to ensure pip works in addition to uv pip.

Tests

CI and manual tests

Checklist

Before submitting this PR, please make sure (put X in square brackets):

  • [x] I have performed a self-review of my code. For an optional AI review, add the gemini-review label.
  • [x] I have necessary comments in my code, particularly in hard-to-understand areas.
  • [x] I have run end-to-end tests tests and provided workload links above if applicable.
  • [x] I have made or will make corresponding changes to the doc if needed, including adding new documentation pages to the relevant Table of Contents (toctree directive) as explained in our documentation.

SamuelMarks avatar Nov 14 '25 22:11 SamuelMarks