diff --git a/eval.sh b/eval.sh deleted file mode 100644 index 1224f6ebc..000000000 --- a/eval.sh +++ /dev/null @@ -1,7 +0,0 @@ -lerobot-eval \ - --policy.path="/home/jade_choghari/.cache/models/pi05" \ - --env.type=libero \ - --env.task=libero_spatial \ - --eval.batch_size=1 \ - --eval.n_episodes=1 \ - --policy.n_action_steps=10 \ diff --git a/src/lerobot/envs/libero.py b/src/lerobot/envs/libero.py index 4eb447b31..350e20ac2 100644 --- a/src/lerobot/envs/libero.py +++ b/src/lerobot/envs/libero.py @@ -219,7 +219,7 @@ class LiberoEnv(gym.Env): def render(self): raw_obs = self._env.env._get_observations() image = self._format_raw_obs(raw_obs)["pixels"]["image"] - image = image[::-1, ::-1] # flip both H and W for visualization + image = image[::-1, ::-1] # flip both H and W for visualization return image def _make_envs_task(self, task_suite: Any, task_id: int = 0): diff --git a/src/lerobot/scripts/lerobot_eval.py b/src/lerobot/scripts/lerobot_eval.py index 7cbd51678..0d66fa1aa 100644 --- a/src/lerobot/scripts/lerobot_eval.py +++ b/src/lerobot/scripts/lerobot_eval.py @@ -764,4 +764,4 @@ def main(): if __name__ == "__main__": - main() \ No newline at end of file + main()