From f7010ff66c38eb788dcbdc2009a9447c4e82de72 Mon Sep 17 00:00:00 2001 From: CarolinePascal Date: Tue, 26 May 2026 18:06:52 +0200 Subject: [PATCH] fix(typo): fixing typo --- src/lerobot/datasets/compute_stats.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/datasets/compute_stats.py b/src/lerobot/datasets/compute_stats.py index fc9a4c562..48b5b0a92 100644 --- a/src/lerobot/datasets/compute_stats.py +++ b/src/lerobot/datasets/compute_stats.py @@ -531,7 +531,7 @@ def compute_episode_stats( ) if features[key]["dtype"] in ["image", "video"]: - normalization_factor = 255.0 if key not in features.depth_keys else 1.0 + normalization_factor = 255.0 if not (features[key].get("info") or {}).get("is_depth_map", False) else 1.0 ep_stats[key] = { k: v if k == "count" else np.squeeze(v / normalization_factor, axis=0) for k, v in ep_stats[key].items() }