Skip to content

Commit 378125e

Browse files
committed
[#141] Add a docstring for core.transforms.generate_transforms()
1 parent ada39ea commit 378125e

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

hlink/linking/core/transforms.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -336,6 +336,16 @@ def generate_transforms(
336336
is_a: bool,
337337
id_col: str,
338338
) -> DataFrame:
339+
"""Generate feature selection columns and return the input dataframe with these new columns attached.
340+
341+
Args:
342+
spark: the Spark session
343+
df_selected: the input Spark DataFrame
344+
feature_selections: a list of feature selections to compute
345+
link_task: the current LinkTask
346+
is_a: whether this is dataset A (True) or dataset B (False)
347+
id_col: the name of the identifier column in the input data frame
348+
"""
339349
not_skipped_feature_selections = [
340350
c
341351
for c in feature_selections

0 commit comments

Comments
 (0)