Fix error reading from generators

mondeja · Jun 7, 2021 · cd7af47 · cd7af47
1 parent 7da82fe
commit cd7af47
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/.bumpversion.cfg b/.bumpversion.cfg
@@ -1,5 +1,5 @@
 [bumpversion]
-current_version = 0.2.4
+current_version = 0.2.5
 
 [bumpversion:file:waves/__init__.py]
 

diff --git a/setup.cfg b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = waves
-version = 0.2.4
+version = 0.2.5
 description = Utility to work with WAV files in a simple way.
 long_description = file: README.rst
 long_description_content_type = text/x-rst

diff --git a/waves/__init__.py b/waves/__init__.py
@@ -7,4 +7,4 @@
 __all__ = ("Sound", "mono_ttf_gen", "stereo_ttf_gen")
 
 __title__ = "waves"
-__version__ = "0.2.4"
+__version__ = "0.2.5"
diff --git a/waves/sound/io.py b/waves/sound/io.py
@@ -553,7 +553,10 @@ def dataframes(self):
                     shape = (self.n_frames, self.n_channels)
                 _data = np.empty(shape, dtype=self.dtype)
                 for i, frame_data in enumerate(self._time_to_frame_generator()):
-                    _data[i] = frame_data
+                    try:
+                        _data[i] = frame_data
+                    except IndexError:
+                        break
                 return _data
         else:
             return self._read_frames()