robocandywrapper 0.2.9__tar.gz → 0.2.11__tar.gz

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

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

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

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

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

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

{robocandywrapper-0.2.9 → robocandywrapper-0.2.11}/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.11}/robocandywrapper/wrapper.py

@@ -642,11 +642,16 @@ class WrappedRobotDataset(torch.utils.data.Dataset):
 
         # Fill missing image keys with zeros/noise
         if self._filled_image_keys:
-            # Find a reference image shape from an existing image key in this item
+            # Find a reference image shape from an existing image key
             ref_shape = None
             for k, v in item.items():
-                if hasattr(v, 'shape') and len(getattr(v, 'shape', ())) >= 3:
-                    ref_shape = v.shape
+                if not hasattr(v, 'shape'):
+                    continue
+                s = v.shape
+                # Match image tensors: 3+ dims, last dim is 3 (RGB) or 4 (RGBA),
+                # and spatial dims are reasonable (> 16 pixels)
+                if len(s) >= 3 and s[-1] in (3, 4) and s[-3] > 16 and s[-2] > 16:
+                    ref_shape = s
                     ref_dtype = v.dtype if hasattr(v, 'dtype') else torch.uint8
                     break
             for key in self._filled_image_keys:

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

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

{robocandywrapper-0.2.9 → robocandywrapper-0.2.11}/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.11",
     description="Sweet wrappers for extending and remixing LeRobot Datasets",
     long_description=long_description,
     long_description_content_type="text/markdown",