Fixed bug in mltu.torch.dataProvider.SequenceHandler, to handle len sequence < queue_size case

Author: pythonlessons

CHANGELOG.md

@@ -1,4 +1,8 @@
-## [1.1.6] - 2022-10-30
+## [1.1.7] - 2022-10-30
+### Changed
+- Fixed bug in `mltu.torch.dataProvider.SequenceHandler`, to handle `len sequence < queue_size` case
+
+# [1.1.6] - 2022-10-30
 ### Changed
 - Fixed dependencies with `librosa` library
 

mltu/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "1.1.6"
+__version__ = "1.1.7"
 
 from .annotations.images import Image
 from .annotations.images import CVImage

mltu/torch/dataProvider.py

@@ -257,9 +257,11 @@ def __exit__(self):
             else:
                 print("Something went wrong")
 
-    def __call__(self, index):
+    def __call__(self, index: int):
         if index == 0:
             for _index in range(self.queue_size):
+                if _index >= self.max_len:
+                    break
                 self.data_queue.put(_index)
 
         while True: