fix assess in DynamicTimeSeriesBucket when max_total_size is used

Author: JanekEbb

lazy_dataset/core.py

@@ -3351,7 +3351,13 @@ def is_completed(self):
 
     def assess(self, example):
         seq_len = self.len_key(example)
-        return self.lower_bound <= seq_len <= self.upper_bound
+        return (
+            (self.lower_bound <= seq_len <= self.upper_bound)
+            and (
+                (self.max_total_size is None)
+                or ((len(self.data) + 1) * max(self.max_len, seq_len) <= self.max_total_size)
+            )
+        )
 
     def _append(self, example):
         super()._append(example)