File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -35,9 +35,10 @@ def image_uri(sagemaker_session):
35
35
36
36
@pytest .fixture (scope = "module" )
37
37
def volume_kms_key (sagemaker_session ):
38
+ role_arn = sagemaker_session .expand_role (ROLE )
38
39
return get_or_create_kms_key (
39
40
sagemaker_session = sagemaker_session ,
40
- role_arn = ROLE ,
41
+ role_arn = role_arn ,
41
42
alias = "integ-test-processing-volume-kms-key-{}" .format (
42
43
sagemaker_session .boto_session .region_name
43
44
),
@@ -46,9 +47,10 @@ def volume_kms_key(sagemaker_session):
46
47
47
48
@pytest .fixture (scope = "module" )
48
49
def output_kms_key (sagemaker_session ):
50
+ role_arn = sagemaker_session .expand_role (ROLE )
49
51
return get_or_create_kms_key (
50
52
sagemaker_session = sagemaker_session ,
51
- role_arn = ROLE ,
53
+ role_arn = role_arn ,
52
54
alias = "integ-test-processing-output-kms-key-{}" .format (
53
55
sagemaker_session .boto_session .region_name
54
56
),
You can’t perform that action at this time.
0 commit comments