More changes for clarity

codalab · May 28, 2024 · fae4335 · fae4335
1 parent adf0f9e
commit fae4335
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/codalab/lib/beam/MultiReaderFileStream.py b/codalab/lib/beam/MultiReaderFileStream.py
@@ -80,7 +80,8 @@ def peek(self, index: int, num_bytes):   # type: ignore
 
         with self._lock:
             # Calculate how many new bytes need to be read
-            new_bytes_needed = num_bytes - (max(self._pos) - self._pos[index])
+            new_pos = self._pos[index] + num_bytes
+            new_bytes_needed = new_pos - max(self._pos)
             if new_bytes_needed > 0:
                 self._fill_buf_bytes(new_bytes_needed)