File tree Expand file tree Collapse file tree 2 files changed +0
-28
lines changed Expand file tree Collapse file tree 2 files changed +0
-28
lines changed Original file line number Diff line number Diff line change @@ -5668,7 +5668,6 @@ def get_caller_identity_arn(self):
5668
5668
instance_name = metadata .get ("ResourceName" )
5669
5669
domain_id = metadata .get ("DomainId" )
5670
5670
user_profile_name = metadata .get ("UserProfileName" )
5671
- space_name = metadata .get ("SpaceName" )
5672
5671
execution_role_arn = metadata .get ("ExecutionRoleArn" )
5673
5672
try :
5674
5673
if domain_id is None :
@@ -5681,11 +5680,6 @@ def get_caller_identity_arn(self):
5681
5680
if execution_role_arn is not None :
5682
5681
return execution_role_arn
5683
5682
5684
- # In Shared Space app, find execution role from DefaultSpaceSettings on domain level
5685
- if space_name is not None :
5686
- domain_desc = self .sagemaker_client .describe_domain (DomainId = domain_id )
5687
- return domain_desc ["DefaultSpaceSettings" ]["ExecutionRole" ]
5688
-
5689
5683
user_profile_desc = self .sagemaker_client .describe_user_profile (
5690
5684
DomainId = domain_id , UserProfileName = user_profile_name
5691
5685
)
Original file line number Diff line number Diff line change @@ -723,28 +723,6 @@ def test_get_caller_identity_arn_from_metadata_file_for_space(boto_session):
723
723
assert actual == expected_role
724
724
725
725
726
- @patch (
727
- "six.moves.builtins.open" ,
728
- mock_open (
729
- read_data = '{"ResourceName": "SageMakerInstance", '
730
- '"DomainId": "d-kbnw5yk6tg8j", '
731
- '"SpaceName": "space_name"}'
732
- ),
733
- )
734
- @patch ("os.path.exists" , side_effect = mock_exists (NOTEBOOK_METADATA_FILE , True ))
735
- def test_get_caller_identity_arn_from_describe_domain_for_space (boto_session ):
736
- sess = Session (boto_session )
737
- expected_role = "arn:aws:iam::369233609183:role/service-role/SageMakerRole-20171129T072388"
738
- sess .sagemaker_client .describe_domain .return_value = {
739
- "DefaultSpaceSettings" : {"ExecutionRole" : expected_role }
740
- }
741
-
742
- actual = sess .get_caller_identity_arn ()
743
-
744
- assert actual == expected_role
745
- sess .sagemaker_client .describe_domain .assert_called_once_with (DomainId = "d-kbnw5yk6tg8j" )
746
-
747
-
748
726
@patch (
749
727
"six.moves.builtins.open" ,
750
728
mock_open (read_data = '{"ResourceName": "SageMakerInstance"}' ),
You can’t perform that action at this time.
0 commit comments