Skip to content
Snippets Groups Projects
Commit 3c67513d 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 8577177a
No related branches found
No related tags found
1 merge request!54Improvements from integration of Hiopy into the Global Hackathon Atmo-only Runs
......@@ -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