@@ -146,7 +146,7 @@ def __init__(
146146 code_location : Optional [str ] = None ,
147147 entry_point : Optional [str ] = None ,
148148 dependencies : Optional [List [Union [str ]]] = None ,
149- instance_groups = None ,
149+ instance_groups : Optional [ Dict [ str , Union [ str , int ]]] = None ,
150150 ** kwargs ,
151151 ):
152152 """Initialize an ``EstimatorBase`` instance.
@@ -2140,7 +2140,7 @@ def __init__(
21402140 code_location : Optional [str ] = None ,
21412141 entry_point : Optional [str ] = None ,
21422142 dependencies : Optional [List [str ]] = None ,
2143- instance_groups = None ,
2143+ instance_groups : Optional [ Dict [ str , Union [ str , int ]]] = None ,
21442144 ** kwargs ,
21452145 ):
21462146 """Initialize an ``Estimator`` instance.
@@ -2429,7 +2429,6 @@ def __init__(
24292429 role ,
24302430 instance_count ,
24312431 instance_type ,
2432- instance_groups ,
24332432 volume_size ,
24342433 volume_kms_key ,
24352434 max_run ,
@@ -2465,6 +2464,7 @@ def __init__(
24652464 entry_point = entry_point ,
24662465 dependencies = dependencies ,
24672466 hyperparameters = hyperparameters ,
2467+ instance_groups = instance_groups ,
24682468 ** kwargs ,
24692469 )
24702470
0 commit comments