Skip to content

Commit 8a07011

Browse files
authored
fix: add region support for invoke client (#71)
1 parent bcb88ca commit 8a07011

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/emd/sdk/invoke/invoker_base.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
from emd.utils.aws_service_utils import get_model_stack_info
1+
from emd.utils.aws_service_utils import get_model_stack_info, get_current_region
22
from emd.models import Model
33
from emd.constants import MODEL_DEFAULT_TAG
44
from emd.models.utils.constants import ServiceType
@@ -45,10 +45,10 @@ def initialize(self):
4545
def get_service_client(self,service_type:str):
4646
if service_type in [ServiceType.SAGEMAKER,ServiceType.SAGEMAKER_ASYNC]:
4747
from emd.sdk.clients.sagemaker_client import SageMakerClient
48-
return SageMakerClient(model_stack_name=self.model_stack_name)
48+
return SageMakerClient(model_stack_name=self.model_stack_name, region_name=get_current_region())
4949
elif service_type == ServiceType.ECS:
5050
from emd.sdk.clients.ecs_client import ECSClient
51-
return ECSClient(model_stack_name=self.model_stack_name)
51+
return ECSClient(model_stack_name=self.model_stack_name, region_name=get_current_region())
5252
else:
5353
raise ValueError(f"Service type {service_type} is not supported")
5454

0 commit comments

Comments
 (0)