Skip to content

Commit e9ac9fa

Browse files
committed
support passing step object to tuning step depends on list
1 parent c0066ea commit e9ac9fa

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/workflow/steps.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -465,7 +465,7 @@ def __init__(
465465
inputs=None,
466466
job_arguments: List[str] = None,
467467
cache_config: CacheConfig = None,
468-
depends_on: List[str] = None,
468+
depends_on: Union[List[str], List[Step]] = None,
469469
):
470470
"""Construct a TuningStep, given a `HyperparameterTuner` instance.
471471
@@ -505,8 +505,8 @@ def __init__(
505505
job_arguments (List[str]): A list of strings to be passed into the processing job.
506506
Defaults to `None`.
507507
cache_config (CacheConfig): A `sagemaker.workflow.steps.CacheConfig` instance.
508-
depends_on (List[str]): A list of step names this `sagemaker.workflow.steps.ProcessingStep`
509-
depends on
508+
depends_on (List[str] or List[Step]): A list of step names or step instance
509+
this `sagemaker.workflow.steps.ProcessingStep` depends on
510510
"""
511511
super(TuningStep, self).__init__(name, StepTypeEnum.TUNING, depends_on)
512512
self.tuner = tuner

0 commit comments

Comments
 (0)