From 31aa0766a2ba517dfdf38845b497894182c6f466 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Wed, 24 Jun 2026 17:46:43 +0200 Subject: [PATCH] fix(compress): removing rerun compression for DepthImages --- src/lerobot/utils/visualization_utils.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/lerobot/utils/visualization_utils.py b/src/lerobot/utils/visualization_utils.py index 924f056fd..5a0f74378 100644 --- a/src/lerobot/utils/visualization_utils.py +++ b/src/lerobot/utils/visualization_utils.py @@ -109,9 +109,7 @@ def log_rerun_data( else: if arr.shape[-1] == 1: img_entity = ( - rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis).compress() - if compress_images - else rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis) + rr.DepthImage(arr, colormap=rr.components.Colormap.Viridis) ) else: img_entity = rr.Image(arr).compress() if compress_images else rr.Image(arr)