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