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

fix(typo): fixing typo

Browse Source
This commit is contained in:
CarolinePascal
2026-05-26 18:06:52 +02:00
parent f7ee453de7
commit f7010ff66c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/compute_stats.py
+1 -1
View File
@@ -531,7 +531,7 @@ def compute_episode_stats(
) )
if features[key]["dtype"] in ["image", "video"]: 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] = { ep_stats[key] = {
k: v if k == "count" else np.squeeze(v / normalization_factor, axis=0) for k, v in ep_stats[key].items() k: v if k == "count" else np.squeeze(v / normalization_factor, axis=0) for k, v in ep_stats[key].items()
} }