diff --git a/python/dolma/core/binning.py b/python/dolma/core/binning.py index 3afee92f..9722ffd9 100644 --- a/python/dolma/core/binning.py +++ b/python/dolma/core/binning.py @@ -232,7 +232,7 @@ def add(self, values: Union[List[Union[int, float]], Union[int, float]], counts: for value, count in zip(values, counts): self._add(value, count) - self._total += 1 + self._total += count self._sum += value * count def add_summary(self, summary: SummaryTuple):