MNT: fix matplotlib deprecation in example

theOehrly · Jan 1, 2024 · 41075e2 · 41075e2
1 parent 92aa85f
commit 41075e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/plot_gear_shifts_on_track.py b/examples/plot_gear_shifts_on_track.py
@@ -10,7 +10,7 @@
 
 import matplotlib.pyplot as plt
 from matplotlib.collections import LineCollection
-from matplotlib import cm
+from matplotlib import colormaps
 import numpy as np
 
 
@@ -37,7 +37,7 @@
 # Create a line collection. Set a segmented colormap and normalize the plot
 # to full integer values of the colormap
 
-cmap = cm.get_cmap('Paired')
+cmap = colormaps['Paired']
 lc_comp = LineCollection(segments, norm=plt.Normalize(1, cmap.N+1), cmap=cmap)
 lc_comp.set_array(gear)
 lc_comp.set_linewidth(4)