Fix stats_factory

huggingface · Feb 13, 2025 · 95ea0ec · 95ea0ec
1 parent f0cfd64
commit 95ea0ec
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py
@@ -141,13 +141,15 @@ def _create_stats(
                     "mean": np.full((3, 1, 1), 0.5, dtype=np.float32).tolist(),
                     "min": np.full((3, 1, 1), 0, dtype=np.float32).tolist(),
                     "std": np.full((3, 1, 1), 0.25, dtype=np.float32).tolist(),
+                    "count": [10],
                 }
             else:
                 stats[key] = {
                     "max": np.full(shape, 1, dtype=dtype).tolist(),
                     "mean": np.full(shape, 0.5, dtype=dtype).tolist(),
                     "min": np.full(shape, 0, dtype=dtype).tolist(),
                     "std": np.full(shape, 0.25, dtype=dtype).tolist(),
+                    "count": [10],
                 }
         return stats