Skip to content

Commit 4072964

Browse files
committed
fix: remove unnecessary get caller identity call
1 parent 4955d22 commit 4072964

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
@@ -390,6 +390,7 @@ def _add_role_to_kwargs(kwargs: JumpStartEstimatorInitKwargs) -> JumpStartEstima
390390
field_name="role",
391391
field_val=kwargs.role,
392392
sagemaker_session=kwargs.sagemaker_session,
393+
default_value=kwargs.role,
393394
)
394395

395396
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)