From 9781d0c52b9d0b2c6be7a4ed98ef7730d3d0e77c Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Wed, 8 Mar 2023 10:53:58 +0000 Subject: [PATCH] Treat (0, None) as fixed dim if is the first dimension --- merlin/dtypes/shape.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/merlin/dtypes/shape.py b/merlin/dtypes/shape.py index db37b643f..c7fec05a7 100644 --- a/merlin/dtypes/shape.py +++ b/merlin/dtypes/shape.py @@ -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)