Skip to content

Commit 74bd774

Browse files
committed
only update non-empty chunks offset
1 parent 8164bfb commit 74bd774

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/zarr/codecs/sharding.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -265,7 +265,8 @@ async def finalize(
265265
) -> Buffer:
266266
index_bytes = await index_encoder(self.index)
267267
if index_location == ShardingCodecIndexLocation.start:
268-
self.index.offsets_and_lengths[..., 0] += len(index_bytes)
268+
empty_chunks_mask = self.index.offsets_and_lengths[..., 0] == MAX_UINT_64
269+
self.index.offsets_and_lengths[~empty_chunks_mask, 0] += len(index_bytes)
269270
index_bytes = await index_encoder(self.index) # encode again with corrected offsets
270271
out_buf = index_bytes + self.buf
271272
else:

0 commit comments

Comments
 (0)