Remove specifying is_ragged in LocalExecutor _transform_data (#173)

NVIDIA-Merlin · Nov 22, 2022 · 025f10d · 025f10d
1 parent 72f967a
commit 025f10d
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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