Skip to content

Commit f22de73

Browse files
committed
fix: Update JsonGet func to correct func
1 parent 4567249 commit f22de73

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/integ/test_workflow.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -67,6 +67,7 @@
6767
ConditionLessThanOrEqualTo,
6868
)
6969
from sagemaker.workflow.condition_step import ConditionStep
70+
from sagemaker.workflow.condition_step import JsonGet as ConditionStepJsonGet
7071
from sagemaker.workflow.callback_step import (
7172
CallbackStep,
7273
CallbackOutput,
@@ -2831,7 +2832,7 @@ def test_end_to_end_pipeline_successful_execution(
28312832

28322833
# define condition step
28332834
cond_lte = ConditionLessThanOrEqualTo(
2834-
left=JsonGet(
2835+
left=ConditionStepJsonGet(
28352836
step=step_eval,
28362837
property_file=evaluation_report,
28372838
json_path="regression_metrics.mse.value",

0 commit comments

Comments
 (0)