From 025f10da5f96c9ae470ee31133801da05fabfbf9 Mon Sep 17 00:00:00 2001 From: Oliver Holworthy Date: Tue, 22 Nov 2022 17:45:46 +0000 Subject: [PATCH] Remove specifying `is_ragged` in LocalExecutor `_transform_data` (#173) --- merlin/dag/executors.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/merlin/dag/executors.py b/merlin/dag/executors.py index 48a5276f0..eab964c8a 100644 --- a/merlin/dag/executors.py +++ b/merlin/dag/executors.py @@ -182,9 +182,7 @@ def _transform_data(self, node, input_data, capture_dtypes=False): elif hasattr(col_series, "numpy"): col_dtype = col_series[0].cpu().numpy().dtype - output_data_schema = output_col_schema.with_dtype( - col_dtype, is_list=is_list, is_ragged=is_list - ) + output_data_schema = output_col_schema.with_dtype(col_dtype, is_list=is_list) if capture_dtypes: node.output_schema.column_schemas[col_name] = output_data_schema