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

…ncy image size
diambra · Apr 30, 2024 · c9cb229 · c9cb229
1 parent a10a9f3
commit c9cb229
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",