allow weights to be single numbers like values

scikit-hep · Dec 24, 2020 · c3789b0 · c3789b0
1 parent c1d28d2
commit c3789b0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/coffea/hist/hist_tools.py b/coffea/hist/hist_tools.py
@@ -954,6 +954,8 @@ def fill(self, **values):
         weight = values.pop("weight", None)
         if isinstance(weight, (awkward1.Array, np.ndarray)):
             weight = coffea.util._ensure_flat(weight)
+        if isinstance(weight, numbers.Number):
+            weight = np.atleast_1d(weight)
         if not all(d.name in values for d in self._axes):
             missing = ", ".join(d.name for d in self._axes if d.name not in values)
             raise ValueError("Not all axes specified for %r.  Missing: %s" % (self, missing))