fix: prematurely exhausting all generator arguments passed to hist plot.

scikit-hep · Oct 3, 2023 · ed3ace3 · ed3ace3
1 parent 826b1aa
commit ed3ace3
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: