From 27b482adf7251a4743bbff72e8a702498e39f72d Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Wed, 10 Jun 2026 19:50:26 +0200 Subject: [PATCH] chore(simplification): removing no longer needed reshape --- src/lerobot/utils/visualization_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/utils/visualization_utils.py b/src/lerobot/utils/visualization_utils.py index 1ff99a482..46fd2c51c 100644 --- a/src/lerobot/utils/visualization_utils.py +++ b/src/lerobot/utils/visualization_utils.py @@ -140,7 +140,7 @@ def log_rerun_data( if arr.ndim == 3 and arr.shape[0] in (1, 3, 4) and arr.shape[-1] not in (1, 3, 4): arr = np.transpose(arr, (1, 2, 0)) if arr.ndim == 1: - rr.log(key, rr.Scalars(arr.reshape(-1).astype(float))) + rr.log(key, rr.Scalars(arr.astype(float))) observation_paths.add(key) else: img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)