Skip to content

Commit 92a8eea

Browse files
author
Brock Wade
committed
fix: add region_name to test
1 parent 2531a1c commit 92a8eea

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

tests/integ/sagemaker/workflow/test_workflow.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1172,7 +1172,13 @@ def walk():
11721172

11731173

11741174
def test_caching_behavior(
1175-
pipeline_session, role, cpu_instance_type, pipeline_name, script_dir, athena_dataset_definition
1175+
pipeline_session,
1176+
role,
1177+
cpu_instance_type,
1178+
pipeline_name,
1179+
script_dir,
1180+
athena_dataset_definition,
1181+
region_name,
11761182
):
11771183
default_bucket = pipeline_session.default_bucket()
11781184
data_path = os.path.join(DATA_DIR, "workflow")
@@ -1475,7 +1481,7 @@ def test_multi_step_framework_processing_pipeline_uploads(
14751481
bucket=default_bucket, key_prefix=expected_prefix
14761482
)
14771483

1478-
# verify
1484+
# verify all distinct artifacts were uploaded
14791485
assert expected_source_dir_tar in s3_code_objects
14801486
assert expected_query_step_artifact in s3_code_objects
14811487
assert expected_prepare_step_artifact in s3_code_objects

0 commit comments

Comments
 (0)