Skip to content

Commit 7dcc1f6

Browse files
committed
Update LocalSagemakerRuntimeClient
1 parent f9a35d9 commit 7dcc1f6

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

src/sagemaker/local/local_session.py

Lines changed: 17 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -540,7 +540,21 @@ def __init__(self, config=None):
540540
self.http = urllib3.PoolManager()
541541
self.serving_port = 8080
542542
self.config = config
543-
self.serving_port = get_config_value("local.serving_port", config) or 8080
543+
544+
@property
545+
def config(self) -> dict:
546+
"""Local config getter"""
547+
return self._config
548+
549+
@config.setter
550+
def config(self, value: dict):
551+
"""Local config setter, this method also updates the `serving_port` attribute.
552+
553+
Args:
554+
value (dict): the new config value
555+
"""
556+
self._config = value
557+
self.serving_port = get_config_value("local.serving_port", self._config) or 8080
544558

545559
def invoke_endpoint(
546560
self,
@@ -688,6 +702,7 @@ def _initialize(
688702
)
689703

690704
self.sagemaker_client = LocalSagemakerClient(self)
705+
self.sagemaker_runtime_client = LocalSagemakerRuntimeClient(self.config)
691706

692707
self.local_mode = True
693708
sagemaker_config = kwargs.get("sagemaker_config", None)
@@ -747,7 +762,7 @@ def config(self, value: Dict | None):
747762
self._config = value
748763

749764
# update the runtime client on config changed
750-
self.sagemaker_runtime_client = LocalSagemakerRuntimeClient(self._config)
765+
self.sagemaker_runtime_client.config = self._config
751766

752767
def logs_for_job(self, job_name, wait=False, poll=5, log_type="All"):
753768
"""A no-op method meant to override the sagemaker client.

0 commit comments

Comments
 (0)