Add ray[data] as tpu dependency (#14691)

Signed-off-by: <ricliu@google.com> Signed-off-by: Richard Liu <ricliu@google.com>
2025-03-12 20:13:48 -07:00
parent 165290d357
commit ab426ec9c0
1 changed files with 1 additions and 0 deletions
--- a/requirements/tpu.txt
+++ b/requirements/tpu.txt
@@ -9,6 +9,7 @@ setuptools-scm>=8
 wheel
 jinja2
 ray[default]
+ray[data]

 # Install torch_xla
 --pre