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

fix(stats): handle scalar stats robustly

Browse Source 
- Wrap cast_stats_to_numpy with np.atleast_1d to prevent 0-d arrays
from scalar stats causing shape mismatches downstream.
This commit is contained in:
Khalil Meftah
2026-06-15 12:28:18 +02:00
parent 949f4fcbe9
commit 0efa3dc874
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/io_utils.py
+1 -1
View File
@@ -153,7 +153,7 @@ def cast_stats_to_numpy(stats: dict) -> dict[str, dict[str, np.ndarray]]:
Returns: Returns:
dict: The statistics dictionary with values cast to numpy arrays. dict: The statistics dictionary with values cast to numpy arrays.
""" """
stats = {key: np.array(value) for key, value in flatten_dict(stats).items()} stats = {key: np.atleast_1d(np.array(value)) for key, value in flatten_dict(stats).items()}
return unflatten_dict(stats) return unflatten_dict(stats)