Skip to content

Commit

Permalink
Treat (0, None) as fixed dim if is the first dimension
Browse files Browse the repository at this point in the history
  • Loading branch information
oliverholworthy committed Mar 8, 2023
1 parent 5494898 commit 8a1599e
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion merlin/dtypes/shape.py
Original file line number Diff line number Diff line change
Expand Up @@ -153,7 +153,7 @@ def __post_init__(self):
if self.dims is not None:
new_dims = []
for i, dim in enumerate(self.dims):
if i == 0 and dim is None:
if i == 0 and dim is None or dim == (0, None):
dim = -1
new_dim = Dimension(dim)
new_dims.append(new_dim)
Expand Down

0 comments on commit 8a1599e

Please sign in to comment.