chore: Unbreak Ray 2.33 unit tests

PiperOrigin-RevId: 678324906
googleapis · Sep 24, 2024 · c669d33 · c669d33
1 parent f89df1f
commit c669d33
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -129,7 +129,7 @@
 ray_testing_extra_require = ray_extra_require + [
     "pytest-xdist",
     # ray train extras required for prediction tests
-    "ray[train] == 2.9.3",
+    "ray[train]",
     # Framework version constraints copied from testing_extra_require
     "scikit-learn",
     "tensorflow",