Fix matplotlib colormap (#225)

* Fix matplotlib colormap * Add changelog entry * Correct changelog entry
cbrnr · Jun 21, 2024 · f457662 · f457662
1 parent 94a85bb
commit f457662
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,4 +1,6 @@
 ## [UNRELEASED] - YYYY-MM-DD
+### Fixed
+- Fix `plot_ecg` for newer versions of Matplotlib ([#225](https://github.com/cbrnr/sleepecg/pull/225) by [Clemens Brunner](https://github.com/cbrnr))
 
 ## [0.5.8] - 2024-05-21
 ### Changed

diff --git a/src/sleepecg/plot.py b/src/sleepecg/plot.py
@@ -63,7 +63,6 @@ def plot_ecg(
     given by `heartbeats`.
     """
     import matplotlib.pyplot as plt
-    from matplotlib.cm import get_cmap
 
     t = np.arange(0, len(ecg) / fs, 1 / fs)
     fig, ax = plt.subplots()
@@ -72,7 +71,7 @@ def plot_ecg(
 
     ax.spines["top"].set_visible(False)
     ax.spines["right"].set_visible(False)
-    colors = cycle(get_cmap("tab10").colors)
+    colors = cycle(plt.get_cmap("tab10").colors)
     markers = cycle(("*", "o", "s", "D", "v", "<", ">", "^", "X", "p"))
     for label, pos, color, marker in zip(kwargs.keys(), kwargs.values(), colors, markers):
         ax.plot(