File tree 2 files changed +5
-4
lines changed
2 files changed +5
-4
lines changed Original file line number Diff line number Diff line change 45
45
46
46
SLEEP_TIME_SECONDS = 1
47
47
SLEEP_TIME_TWO_SECONDS = 2
48
- STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline17 "
49
- STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint17 "
50
- STATIC_MODEL_PACKAGE_GROUP_NAME = "SdkIntegTestStaticPipeline17ModelPackageGroup "
48
+ STATIC_PIPELINE_NAME = "SdkIntegTestStaticPipeline20 "
49
+ STATIC_ENDPOINT_NAME = "SdkIntegTestStaticEndpoint20 "
50
+ STATIC_MODEL_PACKAGE_GROUP_NAME = "SdkIntegTestStaticPipeline20ModelPackageGroup "
51
51
52
52
53
53
@pytest .fixture
Original file line number Diff line number Diff line change 67
67
ConditionLessThanOrEqualTo ,
68
68
)
69
69
from sagemaker .workflow .condition_step import ConditionStep
70
+ from sagemaker .workflow .condition_step import JsonGet as ConditionStepJsonGet
70
71
from sagemaker .workflow .callback_step import (
71
72
CallbackStep ,
72
73
CallbackOutput ,
@@ -2831,7 +2832,7 @@ def test_end_to_end_pipeline_successful_execution(
2831
2832
2832
2833
# define condition step
2833
2834
cond_lte = ConditionLessThanOrEqualTo (
2834
- left = JsonGet (
2835
+ left = ConditionStepJsonGet (
2835
2836
step = step_eval ,
2836
2837
property_file = evaluation_report ,
2837
2838
json_path = "regression_metrics.mse.value" ,
You can’t perform that action at this time.
0 commit comments