fix: prematurely exhausting all generator arguments passed to hist pl…

…ot. (#449)
scikit-hep · Oct 4, 2023 · 85d32a3 · 85d32a3
1 parent 826b1aa
commit 85d32a3
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/mplhep/plot.py b/src/mplhep/plot.py
@@ -306,10 +306,9 @@ def iterable_not_string(arg):
  # Check if iterable
  if iterable_not_string(kwargs[kwarg]):
  # Check if tuple of floats or ints (can be used for colors)
- all_entries_numerical = all(
+ if isinstance(kwargs[kwarg], tuple) and all(
  isinstance(x, int) or isinstance(x, float) for x in kwargs[kwarg]
- )
- if isinstance(kwargs[kwarg], tuple) and all_entries_numerical:
+ ):
  for i in range(len(_chunked_kwargs)):
  _chunked_kwargs[i][kwarg] = kwargs[kwarg]
  else: