MAINT Use non-deprecated matplotlib API to get colormap

amueller · Dec 4, 2023 · 8d3c3b0 · 8d3c3b0
1 parent fa7ac29
commit 8d3c3b0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wordcloud/wordcloud.py b/wordcloud/wordcloud.py
@@ -102,8 +102,8 @@ class colormap_color_func(object):
 
     """
     def __init__(self, colormap):
-        import matplotlib.pyplot as plt
-        self.colormap = plt.cm.get_cmap(colormap)
+        from matplotlib import colormaps
+        self.colormap = colormaps.get_cmap(colormap)
 
     def __call__(self, word, font_size, position, orientation,
                  random_state=None, **kwargs):