Skip to content

Commit cd0126b

Browse files
committed
fix integ tests
1 parent 58e74f9 commit cd0126b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/integ/test_model_monitor.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -244,9 +244,10 @@ def output_kms_key(sagemaker_session):
244244

245245
@pytest.fixture(scope="module")
246246
def 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")
257258
def 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
),

0 commit comments

Comments
 (0)