Skip to content

Commit f3353f3

Browse files
fix: remove upsert from test_workflow (#2505)
Co-authored-by: Ahsan Khan <[email protected]>
1 parent d79906e commit f3353f3

File tree

1 file changed

+1
-22
lines changed

1 file changed

+1
-22
lines changed

tests/integ/test_workflow.py

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -460,35 +460,14 @@ def test_one_step_sklearn_processing_pipeline(
460460
# sagemaker entities. However, the jobs created in the steps themselves execute
461461
# under a potentially different role, often requiring access to S3 and other
462462
# artifacts not required to during creation of the jobs in the pipeline steps.
463-
response = pipeline.create(
464-
role, tags=[{"Key": "foo", "Value": "123"}, {"Key": "bar", "Value": "456"}]
465-
)
463+
response = pipeline.create(role)
466464
create_arn = response["PipelineArn"]
467465
assert re.match(
468466
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",
469467
create_arn,
470468
)
471-
original_tags = sagemaker_session.sagemaker_client.list_tags(ResourceArn=create_arn)
472-
for tag in [{"Key": "foo", "Value": "123"}, {"Key": "bar", "Value": "456"}]:
473-
assert tag in original_tags["Tags"]
474469

475470
pipeline.parameters = [ParameterInteger(name="InstanceCount", default_value=1)]
476-
response = pipeline.upsert(
477-
role, tags=[{"Key": "foo", "Value": "abc"}, {"Key": "baz", "Value": "789"}]
478-
)
479-
update_arn = response["PipelineArn"]
480-
assert re.match(
481-
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",
482-
update_arn,
483-
)
484-
updated_tags = sagemaker_session.sagemaker_client.list_tags(ResourceArn=create_arn)
485-
for tag in [
486-
{"Key": "foo", "Value": "abc"},
487-
{"Key": "bar", "Value": "456"},
488-
{"Key": "baz", "Value": "789"},
489-
]:
490-
assert tag in updated_tags["Tags"]
491-
492471
execution = pipeline.start(parameters={})
493472
assert re.match(
494473
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}/execution/",

0 commit comments

Comments
 (0)