Skip to content

Commit 85770d8

Browse files
committed
fix existing unit test error with sagemaker-train evaluate test execution 2
1 parent 0742bf6 commit 85770d8

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

sagemaker-train/tests/unit/train/evaluate/test_execution.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -254,7 +254,7 @@ def test_extract_with_exception(self):
254254
class 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):
764764
class 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

Comments
 (0)