@@ -254,7 +254,7 @@ def test_extract_with_exception(self):
254254class TestGetOrCreatePipeline :
255255 """Tests for _get_or_create_pipeline function."""
256256
257- @patch ("sagemaker.train.evaluate.execution.Tag " )
257+ @patch ("sagemaker.train.evaluate.execution.ResourceTag " )
258258 @patch ("sagemaker.train.evaluate.execution.Pipeline" )
259259 def test_get_existing_pipeline_and_update (self , mock_pipeline_class , mock_tag_class , mock_session ):
260260 """Test getting and updating existing pipeline via Pipeline.get_all with prefix."""
@@ -764,7 +764,7 @@ def test_get_execution_generic_exception(self, mock_pe_class, mock_session):
764764class TestEvaluationPipelineExecutionGetAll :
765765 """Tests for EvaluationPipelineExecution.get_all() method."""
766766
767- @patch ("sagemaker.train.evaluate.execution.Tag " )
767+ @patch ("sagemaker.train.evaluate.execution.ResourceTag " )
768768 @patch ("sagemaker.train.evaluate.execution.Pipeline" )
769769 @patch ("sagemaker.train.evaluate.execution.PipelineExecution" )
770770 def test_get_all_executions (self , mock_pe_class , mock_pipeline_class , mock_tag_class , mock_session ):
@@ -807,7 +807,7 @@ def test_get_all_executions(self, mock_pe_class, mock_pipeline_class, mock_tag_c
807807 # Verify PipelineExecution.get_all was called with the pipeline name
808808 mock_pe_class .get_all .assert_called_once ()
809809
810- @patch ("sagemaker.train.evaluate.execution.Tag " )
810+ @patch ("sagemaker.train.evaluate.execution.ResourceTag " )
811811 @patch ("sagemaker.train.evaluate.execution.Pipeline" )
812812 @patch ("sagemaker.train.evaluate.execution.PipelineExecution" )
813813 def test_get_all_multiple_eval_types (self , mock_pe_class , mock_pipeline_class , mock_tag_class , mock_session ):
0 commit comments