Use inifinite props cycler instead of itertools

deep-jkl · deep-jkl · commit 703e3ae27f17 · 2022-10-29T11:03:15.000+02:00
diff --git a/examples/style_sheets/style_sheets_reference.py b/examples/style_sheets/style_sheets_reference.py
@@ -8,7 +8,6 @@
 line plot and histogram,
 
 """
-from itertools import cycle
 
 import numpy as np
 import matplotlib.pyplot as plt
@@ -64,7 +63,7 @@ def plot_colored_circles(ax, prng, nb_samples=15):
     the color cycle, because different styles may have different numbers
     of colors.
     """
-    for sty_dict, j in zip(cycle(plt.rcParams['axes.prop_cycle']),
+    for sty_dict, j in zip(plt.rcParams['axes.prop_cycle'](),
                            range(nb_samples)):
         ax.add_patch(plt.Circle(prng.normal(scale=3, size=2),
                                 radius=1.0, color=sty_dict['color']))
