File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 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
2
2
from emd .models import Model
3
3
from emd .constants import MODEL_DEFAULT_TAG
4
4
from emd .models .utils .constants import ServiceType
@@ -45,10 +45,10 @@ def initialize(self):
45
45
def get_service_client (self ,service_type :str ):
46
46
if service_type in [ServiceType .SAGEMAKER ,ServiceType .SAGEMAKER_ASYNC ]:
47
47
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 () )
49
49
elif service_type == ServiceType .ECS :
50
50
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 () )
52
52
else :
53
53
raise ValueError (f"Service type { service_type } is not supported" )
54
54
You can’t perform that action at this time.
0 commit comments