matplotlib fix

Co-Authored-By: Peder Bergebakken Sundt <[email protected]>
aresio · Nov 20, 2023 · 05f8bcf · 05f8bcf
1 parent d3661ea
commit 05f8bcf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/simpful/simpful.py b/simpful/simpful.py
@@ -131,12 +131,12 @@ def draw(self, ax, TGT=None, element=None, highlight=None, xscale="linear"):
             if isinstance(fs, SingletonsSet):
                 xs = [pair[0] for pair in fs._funpointer._pairs]
                 ys = [pair[1] for pair in fs._funpointer._pairs]
-                ax.vlines(x=xs, ymin=0.0, ymax=ys, linestyles=linestyles[nn%4], color=next(ax._get_lines.prop_cycler)['color'], label=fs._term)
+                ax.vlines(x=xs, ymin=0.0, ymax=ys, linestyles=linestyles[nn%4], color=ax._get_lines.get_next_color(), label=fs._term)
 
 
             elif fs._type == "function":
                 y = [fs.get_value(xx) for xx in x]
-                color = next(ax._get_lines.prop_cycler)['color']
+                color = ax._get_lines.get_next_color()
                 lw = 1
 
                 if highlight==fs._term: