fix(typo): fixing typo

This commit is contained in:
CarolinePascal
2026-05-26 18:06:52 +02:00
parent 12a66324fb
commit 56231b17d1
+1 -1
View File
@@ -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()
}