From ab426ec9c04505d311ef222d8609c6eec729248e Mon Sep 17 00:00:00 2001
From: Richard Liu <39319471+richardsliu@users.noreply.github.com>
Date: Wed, 12 Mar 2025 20:13:48 -0700
Subject: [PATCH] Add ray[data] as tpu dependency (#14691)

Signed-off-by: <ricliu@google.com>
Signed-off-by: Richard Liu <ricliu@google.com>
---
 requirements/tpu.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/requirements/tpu.txt b/requirements/tpu.txt
index e8e3b0af5db8c..e071c604b5c0b 100644
--- 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