From 922922652283a03dc1d942cca24e1fc2e01b3e27 Mon Sep 17 00:00:00 2001 From: Remi Date: Thu, 11 Apr 2024 10:35:17 +0200 Subject: [PATCH] Update lerobot/common/envs/utils.py --- lerobot/common/envs/utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lerobot/common/envs/utils.py b/lerobot/common/envs/utils.py index 9d0fb853..4d31ddb2 100644 --- a/lerobot/common/envs/utils.py +++ b/lerobot/common/envs/utils.py @@ -19,6 +19,7 @@ def preprocess_observation(observation, transform=None): img = einops.rearrange(img, "b h w c -> b c h w") obs[imgkey] = img + # TODO(rcadene): enable pixels only baseline with `obs_type="pixels"` in environment by removing requirement for "agent_pos" obs["observation.state"] = torch.from_numpy(observation["agent_pos"]).float() # apply same transforms as in training