Update input_data only if new_columns is not empty in executor (#226

)
NVIDIA-Merlin · Feb 22, 2023 · f5cd9a7 · f5cd9a7
1 parent e3d892e
commit f5cd9a7
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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