diff --git a/tests/integ/sagemaker/lineage/conftest.py b/tests/integ/sagemaker/lineage/conftest.py index 672af41de9..0139a5b658 100644 --- a/tests/integ/sagemaker/lineage/conftest.py +++ b/tests/integ/sagemaker/lineage/conftest.py @@ -45,9 +45,9 @@ SLEEP_TIME_SECONDS = 1 SLEEP_TIME_TWO_SECONDS = 2 -STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline17" -STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint17" -STATIC_MODEL_PACKAGE_GROUP_NAME = "SdkIntegTestStaticPipeline17ModelPackageGroup" +STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline20" +STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint20" +STATIC_MODEL_PACKAGE_GROUP_NAME = "SdkIntegTestStaticPipeline20ModelPackageGroup" @pytest.fixture diff --git a/tests/integ/test_workflow.py b/tests/integ/test_workflow.py index d2c142ee38..160f9f934b 100644 --- a/tests/integ/test_workflow.py +++ b/tests/integ/test_workflow.py @@ -67,6 +67,7 @@ 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, @@ -2831,7 +2832,7 @@ def test_end_to_end_pipeline_successful_execution( # define condition step cond_lte = ConditionLessThanOrEqualTo( - left=JsonGet( + left=ConditionStepJsonGet( step=step_eval, property_file=evaluation_report, json_path="regression_metrics.mse.value",