sdk/python: Memory usage optimization for ObjectFile

Signed-off-by: Aaron Wilson <[email protected]>
NVIDIA · Sep 23, 2024 · e804411 · e804411
1 parent 78dd6c6
commit e804411
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/aistore/sdk/object_file.py b/python/aistore/sdk/object_file.py
@@ -49,7 +49,8 @@ def read(self, size: int = -1) -> bytes:
         if size < 0 or size > len(self):
             size = len(self)
 
-        data = self._buffer[self._pos : self._pos + size]
+        data = memoryview(self._buffer)[self._pos : self._pos + size]
+        # Use len(data) instead of `size` -- Buffer may not contain that much data
         self._pos += len(data)
         return bytes(data)