Merge branch 'release/5.0.1'

pyannote · Jun 14, 2024 · 9126d3e · 9126d3e
2 parents 4b0fd53 + 7b1c268
commit 9126d3e
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/doc/source/changelog.rst b/doc/source/changelog.rst
@@ -2,6 +2,12 @@
 Changelog
 #########
 
+Version 5.0.1 (2024-06-14)
+~~~~~~~~~~~~~~~~~~~~~~~~~~
+
+- fix: fix return type of Timeline.__iter__ (@domsmrz)
+- improve: remove MatplotlibDeprecationWarning (@kimdwkimdw)
+
 Version 5.0.0 (2022-12-15)
 ~~~~~~~~~~~~~~~~~~~~~~~~~~
 

diff --git a/pyannote/core/notebook.py b/pyannote/core/notebook.py
@@ -126,7 +126,7 @@ def __init__(self):
         self.reset()
 
     def reset(self):
-        from matplotlib.cm import get_cmap
+        from matplotlib.pyplot import get_cmap
 
         linewidth = [3, 1]
         linestyle = ["solid", "dashed", "dotted"]

diff --git a/pyannote/core/timeline.py b/pyannote/core/timeline.py
@@ -182,7 +182,7 @@ def __bool__(self):
         """
         return len(self.segments_set_) > 0
 
-    def __iter__(self) -> Iterable[Segment]:
+    def __iter__(self) -> Iterator[Segment]:
         """Iterate over segments (in chronological order)
 
         >>> for segment in timeline: