Merge branch 'develop'

pythonlessons · Oct 30, 2023 · b268e03 · b268e03
2 parents 5465413 + e4e258e
commit b268e03
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 3 deletions.
diff --git a/CHANGELOG.md b/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
 

diff --git a/mltu/__init__.py b/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

diff --git a/mltu/torch/dataProvider.py b/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: