Fix music.Melody.generate_random_melody doctest, forcing pitch_names …

…as list[str]
Jellevanderwerff · Nov 29, 2024 · 93d6817 · 93d6817
1 parent b5c5ad0
commit 93d6817
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/thebeat/music.py b/thebeat/music.py
@@ -820,7 +820,7 @@ class Melody(thebeat.core.sequence.BaseSequence):
     def __init__(
         self,
         rhythm: thebeat.music.Rhythm,
-        pitch_names: npt.NDArray[str] | list[str] | str,
+        pitch_names: npt.NDArray[np.str_] | list[str] | str,
         octave: int | None = None,
         key: str | None = None,
         is_played: list | None = None,
@@ -874,7 +874,7 @@ def __init__(
         else:
             pitch_names_list = pitch_names
 
-        self.pitch_names = pitch_names_list
+        self.pitch_names = [str(pitch_name) for pitch_name in pitch_names_list]
 
         # Add initial events
         self.events = self._make_namedtuples(