admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-15 16:49:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

improve annotation and info dict

Browse Source 
Signed-off-by: Jade Choghari <chogharijade@gmail.com>
This commit is contained in:
Jade Choghari
2025-09-16 14:21:11 +02:00
committed by GitHub
parent d0d9036304
commit 0d1e57f032
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/envs/libero.py
+2 -2
View File
@@ -271,7 +271,7 @@ class LiberoEnv(gym.Env):
info = {"is_success": False} info = {"is_success": False}
return observation, info return observation, info
def step(self, action: np.ndarray) -> tuple[dict[str, Any], dict[str, Any]]: def step(self, action: np.ndarray) -> tuple[dict[str, Any], float, bool, bool, dict[str, Any]]:
if action.ndim != 1: if action.ndim != 1:
raise ValueError( raise ValueError(
f"Expected action to be 1-D (shape (action_dim,)), " f"Expected action to be 1-D (shape (action_dim,)), "
@@ -281,7 +281,7 @@ class LiberoEnv(gym.Env):
is_success = self._env.check_success() is_success = self._env.check_success()
terminated = done or is_success terminated = done or is_success
info["is_success"] = done # is_success info["is_success"] = is_success
observation = self._format_raw_obs(raw_obs) observation = self._format_raw_obs(raw_obs)
if done: if done: