From 95ea0ecdad956eee8a46c5c8d1c2325184cb7593 Mon Sep 17 00:00:00 2001
From: Simon Alibert <simon.alibert@huggingface.co>
Date: Thu, 13 Feb 2025 12:41:45 +0100
Subject: [PATCH] Fix stats_factory

---
 tests/fixtures/dataset_factories.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/fixtures/dataset_factories.py b/tests/fixtures/dataset_factories.py
index fbda025e2..9ae1289b1 100644
--- a/tests/fixtures/dataset_factories.py
+++ b/tests/fixtures/dataset_factories.py
@@ -141,6 +141,7 @@ 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] = {
@@ -148,6 +149,7 @@ def _create_stats(
                     "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