https://github.com/aws/sagemaker-python-sdk/blob/10df8a49f9e149de933eba2910e5bf5a530ebd2f/sagemaker-mlops/src/sagemaker/mlops/workflow/steps.py#L208 Should be: `from sagemaker.mlops.workflow.function_step import DelayedReturn`