diff --git a/merlin/dag/executors.py b/merlin/dag/executors.py index 12f43736e..5b5648830 100644 --- a/merlin/dag/executors.py +++ b/merlin/dag/executors.py @@ -129,8 +129,9 @@ def _build_input_data(self, node, transformable, capture_dtypes=False, validate_ seen_columns = set(parent_output_cols) else: new_columns = set(parent_output_cols) - seen_columns - input_data = concat_columns([input_data, parent_data[list(new_columns)]]) - seen_columns.update(new_columns) + if new_columns: + input_data = concat_columns([input_data, parent_data[list(new_columns)]]) + seen_columns.update(new_columns) # Check for additional input columns that aren't generated by parents # and fetch them from the root DataFrame or DictArray