Skip to content

Commit b110bdd

Browse files
committed
fix integ tests
1 parent 5d9b71c commit b110bdd

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

tests/integ/test_processing.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,9 +35,10 @@ def image_uri(sagemaker_session):
3535

3636
@pytest.fixture(scope="module")
3737
def volume_kms_key(sagemaker_session):
38+
role_arn = sagemaker_session.expand_role(ROLE)
3839
return get_or_create_kms_key(
3940
sagemaker_session=sagemaker_session,
40-
role_arn=ROLE,
41+
role_arn=role_arn,
4142
alias="integ-test-processing-volume-kms-key-{}".format(
4243
sagemaker_session.boto_session.region_name
4344
),
@@ -46,9 +47,10 @@ def volume_kms_key(sagemaker_session):
4647

4748
@pytest.fixture(scope="module")
4849
def output_kms_key(sagemaker_session):
50+
role_arn = sagemaker_session.expand_role(ROLE)
4951
return get_or_create_kms_key(
5052
sagemaker_session=sagemaker_session,
51-
role_arn=ROLE,
53+
role_arn=role_arn,
5254
alias="integ-test-processing-output-kms-key-{}".format(
5355
sagemaker_session.boto_session.region_name
5456
),

0 commit comments

Comments
 (0)