@@ -146,7 +146,7 @@ def __init__(
146
146
code_location : Optional [str ] = None ,
147
147
entry_point : Optional [str ] = None ,
148
148
dependencies : Optional [List [Union [str ]]] = None ,
149
- instance_groups = None ,
149
+ instance_groups : Optional [ Dict [ str , Union [ str , int ]]] = None ,
150
150
** kwargs ,
151
151
):
152
152
"""Initialize an ``EstimatorBase`` instance.
@@ -2140,7 +2140,7 @@ def __init__(
2140
2140
code_location : Optional [str ] = None ,
2141
2141
entry_point : Optional [str ] = None ,
2142
2142
dependencies : Optional [List [str ]] = None ,
2143
- instance_groups = None ,
2143
+ instance_groups : Optional [ Dict [ str , Union [ str , int ]]] = None ,
2144
2144
** kwargs ,
2145
2145
):
2146
2146
"""Initialize an ``Estimator`` instance.
@@ -2429,7 +2429,6 @@ def __init__(
2429
2429
role ,
2430
2430
instance_count ,
2431
2431
instance_type ,
2432
- instance_groups ,
2433
2432
volume_size ,
2434
2433
volume_kms_key ,
2435
2434
max_run ,
@@ -2465,6 +2464,7 @@ def __init__(
2465
2464
entry_point = entry_point ,
2466
2465
dependencies = dependencies ,
2467
2466
hyperparameters = hyperparameters ,
2467
+ instance_groups = instance_groups ,
2468
2468
** kwargs ,
2469
2469
)
2470
2470
0 commit comments