Skip to content

Commit ca0c42f

Browse files
committedMay 2, 2024
Update examples/ibis/feature_engineering/table_dataflow.py
fixes code issue where filter condition cannot be None.
1 parent ba495e7 commit ca0c42f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed
 

‎examples/ibis/feature_engineering/table_dataflow.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -28,4 +28,7 @@ def feature_set(
2828
condition: Optional[ibis.common.deferred.Deferred] = None,
2929
) -> ir.Table:
3030
"""Select feature columns and filter rows"""
31-
return feature_table[feature_selection].filter(condition)
31+
selection = feature_table[feature_selection]
32+
if condition is None:
33+
return selection
34+
return selection.filter(condition)

0 commit comments

Comments
 (0)
Please sign in to comment.