fix: Improve Pipeline workflow unit test branch coverage (aws#2878)

Co-authored-by: Dewen Qi <[email protected]>

Author: 2 people

tests/integ/test_workflow.py

@@ -67,7 +67,6 @@
     ConditionLessThanOrEqualTo,
 )
 from sagemaker.workflow.condition_step import ConditionStep
-from sagemaker.workflow.condition_step import JsonGet as ConditionStepJsonGet
 from sagemaker.workflow.callback_step import (
     CallbackStep,
     CallbackOutput,
@@ -2835,8 +2834,8 @@ def test_end_to_end_pipeline_successful_execution(
 
     # define condition step
     cond_lte = ConditionLessThanOrEqualTo(
-        left=ConditionStepJsonGet(
-            step=step_eval,
+        left=JsonGet(
+            step_name=step_eval.name,
             property_file=evaluation_report,
             json_path="regression_metrics.mse.value",
         ),

tests/unit/sagemaker/workflow/test_functions.py

@@ -233,3 +233,5 @@ def test_add_func_of_json_get():
         json_get_func1 + json_get_func2
 
     assert str(error.value) == "Pipeline variables do not support concatenation."
+=======
+>>>>>>> 0e4fd556 (fix: Improve Pipeline workflow unit test branch coverage (#2878))