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):
244244
245245@pytest .fixture (scope = "module" )
246246def updated_volume_kms_key (sagemaker_session ):
247+ role_arn = sagemaker_session .expand_role (ROLE )
247248 return get_or_create_kms_key (
248249 sagemaker_session = sagemaker_session ,
249- role_arn = ROLE ,
250+ role_arn = role_arn ,
250251 alias = "integ-test-processing-volume-kms-key-updated-{}" .format (
251252 sagemaker_session .boto_session .region_name
252253 ),
@@ -255,9 +256,10 @@ def updated_volume_kms_key(sagemaker_session):
255256
256257@pytest .fixture (scope = "module" )
257258def updated_output_kms_key (sagemaker_session ):
259+ role_arn = sagemaker_session .expand_role (ROLE )
258260 return get_or_create_kms_key (
259261 sagemaker_session = sagemaker_session ,
260- role_arn = ROLE ,
262+ role_arn = role_arn ,
261263 alias = "integ-test-processing-output-kms-key-updated-{}" .format (
262264 sagemaker_session .boto_session .region_name
263265 ),
You can’t perform that action at this time.
0 commit comments