Skip to content

Commit 262a5eb

Browse files
authored
fix: use current session and role when setting config (aws#1493)
* fix: use current session and role when setting config * format
1 parent b07f210 commit 262a5eb

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

src/sagemaker/jumpstart/model.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -355,6 +355,7 @@ def _validate_model_id_and_type():
355355
self.tolerate_deprecated_model = model_init_kwargs.tolerate_deprecated_model
356356
self.region = model_init_kwargs.region
357357
self.sagemaker_session = model_init_kwargs.sagemaker_session
358+
self.role = role
358359
self.config_name = model_init_kwargs.config_name
359360
self.additional_model_data_sources = model_init_kwargs.additional_model_data_sources
360361

@@ -446,6 +447,8 @@ def set_deployment_config(self, config_name: str, instance_type: str) -> None:
446447
model_version=self.model_version,
447448
instance_type=instance_type,
448449
config_name=config_name,
450+
sagemaker_session=self.sagemaker_session,
451+
role=self.role,
449452
)
450453

451454
@property

0 commit comments

Comments
 (0)