Skip to content

Commit 25f9121

Browse files
remove actual emr step execution since we already testing in canary
1 parent 59e4be8 commit 25f9121

File tree

1 file changed

+3
-26
lines changed

1 file changed

+3
-26
lines changed

tests/integ/test_workflow.py

+3-26
Original file line numberDiff line numberDiff line change
@@ -1151,18 +1151,18 @@ def test_two_step_lambda_pipeline_with_output_reference(
11511151

11521152

11531153
def test_two_steps_emr_pipeline(
1154-
sagemaker_session, role, pipeline_name, region_name, emr_cluster_id, emr_script_path
1154+
sagemaker_session, role, pipeline_name, region_name
11551155
):
11561156
instance_count = ParameterInteger(name="InstanceCount", default_value=2)
11571157

11581158
emr_step_config = EMRStepConfig(
11591159
jar="s3://us-west-2.elasticmapreduce/libs/script-runner/script-runner.jar",
1160-
args=[emr_script_path],
1160+
args=["dummy_emr_script_path"],
11611161
)
11621162

11631163
step_emr_1 = EMRStep(
11641164
name="emr-step-1",
1165-
cluster_id=emr_cluster_id,
1165+
cluster_id="j-1YONHTCP3YZKC",
11661166
display_name="emr_step_1",
11671167
description="MyEMRStepDescription",
11681168
step_config=emr_step_config,
@@ -1189,29 +1189,6 @@ def test_two_steps_emr_pipeline(
11891189
assert re.match(
11901190
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}", create_arn
11911191
)
1192-
1193-
execution = pipeline.start()
1194-
try:
1195-
execution.wait(delay=60, max_attempts=5)
1196-
except WaiterError:
1197-
pass
1198-
1199-
execution_steps = execution.list_steps()
1200-
assert len(execution_steps) == 2
1201-
assert execution_steps[0]["StepName"] == "emr-step-1"
1202-
assert execution_steps[0].get("FailureReason", "") == ""
1203-
assert execution_steps[0]["StepStatus"] == "Succeeded"
1204-
assert execution_steps[1]["StepName"] == "emr-step-2"
1205-
assert execution_steps[1].get("FailureReason", "") == ""
1206-
assert execution_steps[1]["StepStatus"] == "Succeeded"
1207-
1208-
pipeline.parameters = [ParameterInteger(name="InstanceCount", default_value=1)]
1209-
response = pipeline.update(role)
1210-
update_arn = response["PipelineArn"]
1211-
assert re.match(
1212-
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",
1213-
update_arn,
1214-
)
12151192
finally:
12161193
try:
12171194
pipeline.delete()

0 commit comments

Comments
 (0)