Remove tensorflow installation from ray rl lib extra to limit depende…

…ncy image size
diambra · Apr 30, 2024 · 9d31fc9 · 9d31fc9
1 parent 0e7e342
commit 9d31fc9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -13,7 +13,7 @@
         "pyyaml",
     ],
     "stable-baselines3": ["stable-baselines3[extra]~=2.1.0", "pyyaml"],
-    "ray-rllib": ["ray[rllib]~=2.7.0", "tensorflow", "torch", "pyyaml"],
+    "ray-rllib": ["ray[rllib]~=2.7.0", "torch", "pyyaml"],
     "sheeprl": [
         "sheeprl==0.5.5",
         "importlib-resources==6.1.0",