From b3290017c9ac9303f2190682d2a8ac5cd0872efb Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Tue, 7 Mar 2023 19:31:22 +0000 Subject: [PATCH] If None provided for first dimension, change to fixed dim --- merlin/dtypes/shape.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/merlin/dtypes/shape.py b/merlin/dtypes/shape.py index d3038c67c..7cde2e52f 100644 --- a/merlin/dtypes/shape.py +++ b/merlin/dtypes/shape.py @@ -153,6 +153,8 @@ 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: + dim = -1 new_dim = Dimension(dim) new_dims.append(new_dim)