From 4027d97c7070484296539a8836502d8dc6f51004 Mon Sep 17 00:00:00 2001 From: mshukor Date: Sun, 1 Jun 2025 09:35:22 +0200 Subject: [PATCH] fix for observations with text --- lerobot/common/robot_devices/control_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/robot_devices/control_utils.py b/lerobot/common/robot_devices/control_utils.py index e2131a6a8..88fe97ea8 100644 --- a/lerobot/common/robot_devices/control_utils.py +++ b/lerobot/common/robot_devices/control_utils.py @@ -109,7 +109,7 @@ def predict_action(observation, policy, device, use_amp): ): # Convert to pytorch format: channel first and float32 in [0,1] with batch dimension for name in observation: - if isinstance(observation[name], list): + if not isinstance(observation[name], torch.Tensor): continue # VLA-like and multirobot policies include textual inputs in the observation if "image" in name: