robocandywrapper 0.2.9__tar.gz → 0.2.10__tar.gz

{robocandywrapper-0.2.9/robocandywrapper.egg-info → robocandywrapper-0.2.10}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robocandywrapper
-Version: 0.2.9
+Version: 0.2.10
 Summary: Sweet wrappers for extending and remixing LeRobot Datasets
 Author: RoboCandyWrapper Contributors
 License: MIT License

{robocandywrapper-0.2.9 → robocandywrapper-0.2.10}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "robocandywrapper"
-version = "0.2.9"
+version = "0.2.10"
 description = "Sweet wrappers for extending and remixing LeRobot Datasets"
 readme = "README.md"
 requires-python = ">=3.10"

{robocandywrapper-0.2.9 → robocandywrapper-0.2.10}/robocandywrapper/__init__.py

@@ -17,7 +17,7 @@ from robocandywrapper.constants import (
     EPISODE_OUTCOME_PLUGIN_NAME,
 )
 
-__version__ = "0.2.9"
+__version__ = "0.2.10"
 
 __all__ = [
     "DatasetPlugin",

{robocandywrapper-0.2.9 → robocandywrapper-0.2.10}/robocandywrapper/metadata_view.py

@@ -79,11 +79,25 @@ def aggregate_stats_weighted(
         if not stats_with_key:
             continue
         
-        # Extract arrays
-        means = np.stack([np.array(s["mean"]) for s in stats_with_key])
-        stds = np.stack([np.array(s["std"]) for s in stats_with_key])
-        mins = np.stack([np.array(s["min"]) for s in stats_with_key])
-        maxs = np.stack([np.array(s["max"]) for s in stats_with_key])
+        # Extract arrays, padding to max dim if shapes differ
+        def _to_arrays(stat_key, pad_value=0.0):
+            arrs = [np.array(s[stat_key]) for s in stats_with_key]
+            if any(a.shape != arrs[0].shape for a in arrs):
+                max_dim = max(a.shape[-1] for a in arrs if a.ndim > 0)
+                padded = []
+                for a in arrs:
+                    if a.ndim > 0 and a.shape[-1] < max_dim:
+                        pad_width = [(0, 0)] * (a.ndim - 1) + [(0, max_dim - a.shape[-1])]
+                        padded.append(np.pad(a, pad_width, constant_values=pad_value))
+                    else:
+                        padded.append(a)
+                return np.stack(padded)
+            return np.stack(arrs)
+
+        means = _to_arrays("mean", pad_value=0.0)
+        stds = _to_arrays("std", pad_value=1.0)
+        mins = _to_arrays("min", pad_value=0.0)
+        maxs = _to_arrays("max", pad_value=0.0)
         
         # Get counts and apply weight multipliers
         # Extract scalar count value (handle both scalar and array counts)

{robocandywrapper-0.2.9 → robocandywrapper-0.2.10/robocandywrapper.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: robocandywrapper
-Version: 0.2.9
+Version: 0.2.10
 Summary: Sweet wrappers for extending and remixing LeRobot Datasets
 Author: RoboCandyWrapper Contributors
 License: MIT License

{robocandywrapper-0.2.9 → robocandywrapper-0.2.10}/setup.py

@@ -9,7 +9,7 @@ long_description = readme_file.read_text(encoding="utf-8") if readme_file.exists
 
 setup(
     name="robocandywrapper",
-    version="0.2.9",
+    version="0.2.10",
     description="Sweet wrappers for extending and remixing LeRobot Datasets",
     long_description=long_description,
     long_description_content_type="text/markdown",