From 26887310643a018190eb13943be41fe40fc68397 Mon Sep 17 00:00:00 2001
From: Jade Choghari <chogharijade@gmail.com>
Date: Thu, 28 Aug 2025 17:59:34 +0300
Subject: [PATCH] Add dep (#4)

* Add 'libero' dependencies to pyproject.toml

* Add Git dependencies for egl_probe and LIBERO

* Update libero-requirements.txt

* add future dep
---
 libero-requirements.txt |  2 ++
 pyproject.toml          | 21 +++++++++++++++++++--
 2 files changed, 21 insertions(+), 2 deletions(-)
 create mode 100644 libero-requirements.txt

diff --git a/libero-requirements.txt b/libero-requirements.txt
new file mode 100644
index 000000000..c4998ed9a
--- /dev/null
+++ b/libero-requirements.txt
@@ -0,0 +1,2 @@
+git+https://github.com/jadechoghari/egl_probe.git#egg=egl_probe
+git+https://github.com/jadechoghari/LIBERO.git@main#egg=libero
diff --git a/pyproject.toml b/pyproject.toml
index 2bc57c076..066d87094 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -134,7 +134,23 @@ video_benchmark = ["scikit-image>=0.23.2", "pandas>=2.2.2"]
 aloha = ["gym-aloha>=0.1.1"]
 pusht = ["gym-pusht>=0.1.5", "pymunk>=6.6.0,<7.0.0"] # TODO: Fix pymunk version in gym-pusht instead
 xarm = ["gym-xarm>=0.1.1"]
-
+libero = [
+    "hydra-core>=1.2,<1.4",
+    "numpy",
+    "wandb",
+    "easydict",
+    "transformers",
+    "opencv-python",
+    "robomimic==0.2.0",
+    "einops",
+    "thop",
+    "robosuite==1.4.0",
+    "bddl==1.0.1",
+    "matplotlib",
+    "cloudpickle",
+    "future",
+    "gym",
+]
 # All
 all = [
     "lerobot[dynamixel]",
@@ -152,7 +168,8 @@ all = [
     "lerobot[video_benchmark]",
     "lerobot[aloha]",
     "lerobot[pusht]",
-    "lerobot[xarm]"
+    "lerobot[xarm]",
+    "lerobot[libero]"
 ]
 
 [project.scripts]