Skip to content

Commit fa3d6f1

Browse files
authored
fix: Remove unnecessary get caller identity call (#3960)
1 parent 7f33995 commit fa3d6f1

File tree

2 files changed

+2
-0
lines changed

2 files changed

+2
-0
lines changed

src/sagemaker/jumpstart/factory/estimator.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -396,6 +396,7 @@ def _add_role_to_kwargs(kwargs: JumpStartEstimatorInitKwargs) -> JumpStartEstima
396396
field_name="role",
397397
field_val=kwargs.role,
398398
sagemaker_session=kwargs.sagemaker_session,
399+
default_value=kwargs.role,
399400
)
400401

401402
return kwargs

src/sagemaker/jumpstart/factory/model.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -127,6 +127,7 @@ def _add_role_to_kwargs(kwargs: JumpStartModelInitKwargs) -> JumpStartModelInitK
127127
field_name="role",
128128
field_val=kwargs.role,
129129
sagemaker_session=kwargs.sagemaker_session,
130+
default_value=kwargs.role,
130131
)
131132

132133
return kwargs

0 commit comments

Comments
 (0)