Skip to content

Commit a3ac65f

Browse files
fix: remove upsert from test_workflow
1 parent f4ad8b2 commit a3ac65f

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
@@ -458,35 +458,14 @@ def test_one_step_sklearn_processing_pipeline(
458458
# sagemaker entities. However, the jobs created in the steps themselves execute
459459
# under a potentially different role, often requiring access to S3 and other
460460
# artifacts not required to during creation of the jobs in the pipeline steps.
461-
response = pipeline.create(
462-
role, tags=[{"Key": "foo", "Value": "123"}, {"Key": "bar", "Value": "456"}]
463-
)
461+
response = pipeline.create(role)
464462
create_arn = response["PipelineArn"]
465463
assert re.match(
466464
fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",
467465
create_arn,
468466
)
469-
original_tags = sagemaker_session.sagemaker_client.list_tags(ResourceArn=create_arn)
470-
for tag in [{"Key": "foo", "Value": "123"}, {"Key": "bar", "Value": "456"}]:
471-
assert tag in original_tags["Tags"]
472467

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

0 commit comments

Comments
 (0)