[#134] Allow any number of input columns in the array feature selection

ipums · May 30, 2024 · 39f118d · 39f118d
1 parent dbb9eee
commit 39f118d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/hlink/linking/core/transforms.py b/hlink/linking/core/transforms.py
@@ -123,9 +123,9 @@ def parse_feature_selections(
             return df_selected
 
         elif transform == "array":
-            col1, col2 = feature_selection["input_columns"]
+            input_cols = feature_selection["input_columns"]
             output_col = feature_selection["output_column"]
-            df_selected = df_selected.withColumn(output_col, array(col1, col2))
+            df_selected = df_selected.withColumn(output_col, array(input_cols))
             return df_selected
 
         elif transform == "union":