From 1fd3b2e2db18daa331879c9b831fbdd44c62ed2c Mon Sep 17 00:00:00 2001 From: Steven Palma Date: Wed, 28 May 2025 15:29:08 +0200 Subject: [PATCH] fix(utils): Convert observation values in predict_action to torch.Tensor (#1157) Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com> --- lerobot/common/utils/control_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lerobot/common/utils/control_utils.py b/lerobot/common/utils/control_utils.py index 012f9f270..327d86653 100644 --- a/lerobot/common/utils/control_utils.py +++ b/lerobot/common/utils/control_utils.py @@ -112,6 +112,7 @@ def predict_action( ): # Convert to pytorch format: channel first and float32 in [0,1] with batch dimension for name in observation: + observation[name] = torch.from_numpy(observation[name]) if "image" in name: observation[name] = observation[name].type(torch.float32) / 255 observation[name] = observation[name].permute(2, 0, 1).contiguous()