Skip to content

Commit 816e50a

Browse files
author
Will Badr
committed
adding support for filters to TransfomStep
1 parent 3ef4a7d commit 816e50a

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/stepfunctions/steps/sagemaker.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -154,9 +154,9 @@ def __init__(self, state_id, transformer, job_name, model_name, data, data_type=
154154
compression_type=compression_type,
155155
split_type=split_type,
156156
job_name=job_name,
157-
input_filter=None,
158-
output_filter=None,
159-
join_source=None
157+
input_filter=input_filter,
158+
output_filter=output_filter,
159+
join_source=join_source
160160
)
161161
else:
162162
parameters = transform_config(
@@ -166,9 +166,9 @@ def __init__(self, state_id, transformer, job_name, model_name, data, data_type=
166166
content_type=content_type,
167167
compression_type=compression_type,
168168
split_type=split_type,
169-
input_filter=None,
170-
output_filter=None,
171-
join_source=None
169+
input_filter=input_filter,
170+
output_filter=output_filter,
171+
join_source=join_source
172172
)
173173

174174
if isinstance(job_name, (ExecutionInput, StepInput)):

0 commit comments

Comments
 (0)