Fix win

xorbitsai · Dec 8, 2022 · a713453 · a713453
1 parent 2d2d5b2
commit a713453
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mars/storage/shared_memory.py b/mars/storage/shared_memory.py
@@ -79,6 +79,7 @@ def _read_init(self):
         self._buffer = self._mv = shm.buf
         if self._size is None:
             (self._size,) = _qword_pack.unpack(shm.buf[-8:])
+        self._buffer = self._buffer[: self._size + 8]
 
     def _write_close(self):
         pass
@@ -88,7 +89,7 @@ def _read_close(self):
 
     def get_buffer(self):
         self.init()
-        return self.shm.buf
+        return self._buffer
 
 
 class ShmStorageFileObject(StorageFileObject):