File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -465,7 +465,7 @@ def __init__(
465
465
inputs = None ,
466
466
job_arguments : List [str ] = None ,
467
467
cache_config : CacheConfig = None ,
468
- depends_on : List [str ] = None ,
468
+ depends_on : Union [ List [str ], List [ Step ] ] = None ,
469
469
):
470
470
"""Construct a TuningStep, given a `HyperparameterTuner` instance.
471
471
@@ -505,8 +505,8 @@ def __init__(
505
505
job_arguments (List[str]): A list of strings to be passed into the processing job.
506
506
Defaults to `None`.
507
507
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
510
510
"""
511
511
super (TuningStep , self ).__init__ (name , StepTypeEnum .TUNING , depends_on )
512
512
self .tuner = tuner
You can’t perform that action at this time.
0 commit comments