Skip to content

Commit 1b46956

Browse files
committed
Fixed Dataset.subset when buffer_size is None
1 parent 109f9b6 commit 1b46956

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

neuralmonkey/dataset.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -616,5 +616,6 @@ def subset(self, start: int, length: int) -> "Dataset":
616616
iterators=slices,
617617
batching=self.batching,
618618
outputs=outputs,
619-
buffer_size=buffer_size,
619+
buffer_size=((self.buffer_min_size, self.buffer_size)
620+
if self.lazy else None),
620621
shuffled=self.shuffled)

0 commit comments

Comments
 (0)