From 39b69efd8ec2d8f81d93e63448257f82c32f1190 Mon Sep 17 00:00:00 2001 From: wadeKeith Date: Tue, 24 Mar 2026 12:33:07 +0800 Subject: [PATCH] docs: clarify Python 3.11 requirement for RLDS group tensorflow-cpu==2.15.0 only provides cp311 wheels on PyPI, so `uv sync --group rlds` fails when the venv uses Python 3.12+. - Add explicit Python 3.11 setup commands to DROID training docs - Add comment in pyproject.toml noting the constraint Closes #907 --- examples/droid/README_train.md | 5 ++++- pyproject.toml | 2 ++ 2 files changed, 6 insertions(+), 1 deletion(-) diff --git a/examples/droid/README_train.md b/examples/droid/README_train.md index c8ad660656..a283dedf25 100644 --- a/examples/droid/README_train.md +++ b/examples/droid/README_train.md @@ -8,8 +8,11 @@ for larger datasets like DROID -- they are working on improving it though). Belo ## Install -We need a few additional dependencies for RLDS data loading. Run: +We need a few additional dependencies for RLDS data loading. The RLDS group depends on `tensorflow-cpu==2.15.0`, which only ships wheels for **Python 3.11**. Make sure you create your virtual environment with Python 3.11 before syncing: + ```bash +uv python install 3.11 +uv venv --python 3.11 uv sync --group rlds ``` diff --git a/pyproject.toml b/pyproject.toml index c4a06e5328..6c8a0f4bde 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -54,6 +54,8 @@ dev = [ "pynvml>=12.0.0", ] rlds = [ + # NOTE: tensorflow-cpu 2.15.0 only ships cp311 wheels on PyPI. + # Use `uv venv --python 3.11` before `uv sync --group rlds`. "dlimp", "tensorflow-cpu==2.15.0", "tensorflow-datasets==4.9.9",