From 6f6e046c531b118038d2192be0e76b77ab382c3e Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Wed, 28 May 2025 20:38:51 +0200 Subject: [PATCH] [skip-ci] feat(audio rerun): adding support for multiple channels visualization with rerun --- 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 05a30df3e..df5c48b27 100644 --- a/src/lerobot/utils/visualization_utils.py +++ b/src/lerobot/utils/visualization_utils.py @@ -166,7 +166,7 @@ def log_rerun_data( ), ) ], - columns=rr.Scalar.columns(scalar=observation[key]), + columns=rr.Scalar.columns(scalar=np.mean(observation[key], axis=1)), ) elif arr.ndim == 3: rr.log(key, rr.Image(arr), static=True)