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)