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 @@ -244,9 +244,10 @@ def output_kms_key(sagemaker_session):
244
244
245
245
@pytest .fixture (scope = "module" )
246
246
def updated_volume_kms_key (sagemaker_session ):
247
+ role_arn = sagemaker_session .expand_role (ROLE )
247
248
return get_or_create_kms_key (
248
249
sagemaker_session = sagemaker_session ,
249
- role_arn = ROLE ,
250
+ role_arn = role_arn ,
250
251
alias = "integ-test-processing-volume-kms-key-updated-{}" .format (
251
252
sagemaker_session .boto_session .region_name
252
253
),
@@ -255,9 +256,10 @@ def updated_volume_kms_key(sagemaker_session):
255
256
256
257
@pytest .fixture (scope = "module" )
257
258
def updated_output_kms_key (sagemaker_session ):
259
+ role_arn = sagemaker_session .expand_role (ROLE )
258
260
return get_or_create_kms_key (
259
261
sagemaker_session = sagemaker_session ,
260
- role_arn = ROLE ,
262
+ role_arn = role_arn ,
261
263
alias = "integ-test-processing-output-kms-key-updated-{}" .format (
262
264
sagemaker_session .boto_session .region_name
263
265
),
You can’t perform that action at this time.
0 commit comments