Skip to content

Commit c8966c7

Browse files
author
Brock Wade
committed
Merge branch 'step-args-bug-fix' into runproc-bug-fix
2 parents 54dd242 + bb9a044 commit c8966c7

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

src/sagemaker/spark/processing.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -316,7 +316,10 @@ def _extend_processing_args(self, inputs, outputs, **kwargs):
316316
self._validate_configuration(configuration)
317317
extended_inputs.append(self._stage_configuration(configuration))
318318

319-
return extended_inputs, extended_outputs
319+
return (
320+
extended_inputs if extended_inputs else None,
321+
extended_outputs if extended_outputs else None,
322+
)
320323

321324
def start_history_server(self, spark_event_logs_s3_uri=None):
322325
"""Starts a Spark history server.

0 commit comments

Comments
 (0)