Skip to content

Commit

Permalink
(fix): type for shape
Browse files Browse the repository at this point in the history
  • Loading branch information
ilan-gold committed Nov 22, 2024
1 parent c89e5e0 commit 545393a
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion tests/test_io_elementwise.py
Original file line number Diff line number Diff line change
Expand Up @@ -74,7 +74,9 @@ def sparse_format(request):
return request.param


def create_dense_store(store: str, *, shape=DEFAULT_SHAPE) -> H5Group | ZarrGroup:
def create_dense_store(
store: str, *, shape: tuple[int, ...] = DEFAULT_SHAPE
) -> H5Group | ZarrGroup:
X = np.random.randn(*shape)

write_elem(store, "X", X)
Expand Down

0 comments on commit 545393a

Please sign in to comment.