From 8a1599ede8c7363b78db209a0c9b04bf0aced7e4 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..d64a90d76 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)