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

chore(simplification): removing no longer needed reshape

Browse Source
This commit is contained in:
CarolinePascal
2026-06-10 19:50:26 +02:00
parent 21d158e066
commit 27b482adf7
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/utils/visualization_utils.py
+1 -1
View File
@@ -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): 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)) arr = np.transpose(arr, (1, 2, 0))
if arr.ndim == 1: 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) observation_paths.add(key)
else: else:
img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr) img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)