Skip to content
Snippets Groups Projects
Commit 41cc2115 authored by Siddhant Tibrewal's avatar Siddhant Tibrewal
Browse files

fixed chunk-shape heuristic in cases of it being specified less than the shape

parent 917ff5c1
No related branches found
No related tags found
No related merge requests found
Pipeline #101136 waiting for manual action
......@@ -86,7 +86,7 @@ def add_variable(
_shard_shape = None
if chunk_shape is not None:
_chunk_shape = tuple(min(chunk_shape, shape))
_chunk_shape = tuple(min(a, b) for a, b in zip(chunk_shape, shape))
if chunks_per_shard is not None:
_shard_shape = tuple(i * chunks_per_shard for i in _chunk_shape)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment