@@ -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