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

scikit-hep · Oct 3, 2023 · 8a7ea58 · 8a7ea58
1 parent ad565e7
commit 8a7ea58
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/src/mplhep/plot.py b/src/mplhep/plot.py
@@ -306,12 +306,11 @@ 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(
-                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]
+            if isinstance(kwargs[kwarg], tuple):
+                all_entries_numerical = all(isinstance(x, int) or isinstance(x, float) for x in kwargs[kwarg])
+                if all_entries_numerical:
+                    for i in range(len(_chunked_kwargs)):
+                        _chunked_kwargs[i][kwarg] = kwargs[kwarg]
             else:
                 for i, kw in enumerate(kwargs[kwarg]):
                     _chunked_kwargs[i][kwarg] = kw