Skip to content

Commit 84f1c45

Browse files
committed
fix integ tests
1 parent b110bdd commit 84f1c45

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
@@ -220,9 +220,10 @@ def byoc_monitoring_schedule_name(sagemaker_session, output_kms_key, volume_kms_
220220

221221
@pytest.fixture(scope="module")
222222
def volume_kms_key(sagemaker_session):
223+
role_arn = sagemaker_session.expand_role(ROLE)
223224
return get_or_create_kms_key(
224225
sagemaker_session=sagemaker_session,
225-
role_arn=ROLE,
226+
role_arn=role_arn,
226227
alias="integ-test-processing-volume-kms-key-{}".format(
227228
sagemaker_session.boto_session.region_name
228229
),
@@ -231,9 +232,10 @@ def volume_kms_key(sagemaker_session):
231232

232233
@pytest.fixture(scope="module")
233234
def output_kms_key(sagemaker_session):
235+
role_arn = sagemaker_session.expand_role(ROLE)
234236
return get_or_create_kms_key(
235237
sagemaker_session=sagemaker_session,
236-
role_arn=ROLE,
238+
role_arn=role_arn,
237239
alias="integ-test-processing-output-kms-key-{}".format(
238240
sagemaker_session.boto_session.region_name
239241
),

0 commit comments

Comments
 (0)