@@ -460,35 +460,14 @@ def test_one_step_sklearn_processing_pipeline(
460
460
# sagemaker entities. However, the jobs created in the steps themselves execute
461
461
# under a potentially different role, often requiring access to S3 and other
462
462
# 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 )
466
464
create_arn = response ["PipelineArn" ]
467
465
assert re .match (
468
466
fr"arn:aws:sagemaker:{ region_name } :\d{{12}}:pipeline/{ pipeline_name } " ,
469
467
create_arn ,
470
468
)
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" ]
474
469
475
470
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
-
492
471
execution = pipeline .start (parameters = {})
493
472
assert re .match (
494
473
fr"arn:aws:sagemaker:{ region_name } :\d{{12}}:pipeline/{ pipeline_name } /execution/" ,
0 commit comments