@@ -540,7 +540,21 @@ def __init__(self, config=None):
540
540
self .http = urllib3 .PoolManager ()
541
541
self .serving_port = 8080
542
542
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
544
558
545
559
def invoke_endpoint (
546
560
self ,
@@ -688,6 +702,7 @@ def _initialize(
688
702
)
689
703
690
704
self .sagemaker_client = LocalSagemakerClient (self )
705
+ self .sagemaker_runtime_client = LocalSagemakerRuntimeClient (self .config )
691
706
692
707
self .local_mode = True
693
708
sagemaker_config = kwargs .get ("sagemaker_config" , None )
@@ -747,7 +762,7 @@ def config(self, value: Dict | None):
747
762
self ._config = value
748
763
749
764
# update the runtime client on config changed
750
- self .sagemaker_runtime_client = LocalSagemakerRuntimeClient ( self ._config )
765
+ self .sagemaker_runtime_client . config = self ._config
751
766
752
767
def logs_for_job (self , job_name , wait = False , poll = 5 , log_type = "All" ):
753
768
"""A no-op method meant to override the sagemaker client.
0 commit comments