From 3d0e431206bb036d788bbbb2f84d695e43162768 Mon Sep 17 00:00:00 2001 From: Manu Compen Date: Mon, 19 Jan 2026 14:49:24 +0100 Subject: [PATCH] minor: fix import path --- sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py b/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py index 295fb633ff..e9bae28732 100644 --- a/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py +++ b/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py @@ -202,7 +202,7 @@ def _find_dependencies_in_step_arguments( else: dependencies.add(self._get_step_name_from_str(referenced_step, step_map)) - from sagemaker.core.workflow.function_step import DelayedReturn + from sagemaker.mlops.workflow.function_step import DelayedReturn # TODO: we can remove the if-elif once move the validators to JsonGet constructor if isinstance(pipeline_variable, JsonGet):