Skip to content

Commit 031db8f

Browse files
asumitamazonjerrypeng7773
authored andcommitted
fix: Moving the newly added field instance_group to the end of method
Moving the newly added field instance_group to the end of method
1 parent fa0a780 commit 031db8f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/sagemaker/estimator.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)