admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

debug

Browse Source
This commit is contained in:
AdilZouitine
2025-09-09 13:35:45 +02:00
parent 5dababd21e
commit 8b43339563
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/envs/utils.py
+1
View File
@@ -151,6 +151,7 @@ def add_envs_task(env: gym.vector.VectorEnv, observation: dict[str, Any]) -> dic
observation["task"] = task_result observation["task"] = task_result
else: # For envs without language instructions, e.g. aloha transfer cube and etc. else: # For envs without language instructions, e.g. aloha transfer cube and etc.
print(f"{observation.keys()}")
num_envs = observation[list(observation.keys())[0]].shape[0] num_envs = observation[list(observation.keys())[0]].shape[0]
observation["task"] = ["" for _ in range(num_envs)] observation["task"] = ["" for _ in range(num_envs)]
return observation return observation