From 0d1e57f032ec667bf05d5dc855c8307ea6d6129d Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Tue, 16 Sep 2025 14:21:11 +0200 Subject: [PATCH] improve annotation and info dict Signed-off-by: Jade Choghari --- src/lerobot/envs/libero.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/envs/libero.py b/src/lerobot/envs/libero.py index 7ec9b34a2..cb24c928a 100644 --- a/src/lerobot/envs/libero.py +++ b/src/lerobot/envs/libero.py @@ -271,7 +271,7 @@ class LiberoEnv(gym.Env): info = {"is_success": False} 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: raise ValueError( f"Expected action to be 1-D (shape (action_dim,)), " @@ -281,7 +281,7 @@ class LiberoEnv(gym.Env): is_success = self._env.check_success() terminated = done or is_success - info["is_success"] = done # is_success + info["is_success"] = is_success observation = self._format_raw_obs(raw_obs) if done: