如何在由步骤函数调用的Lambda函数中抛出TaskFailed错误

class TaskFailed(Exception):
    pass

def lambda_handler(event, context):
    # 执行任务的代码逻辑
    try:
        # 任务执行过程中的代码
        if error_condition:
            raise TaskFailed("任务执行失败")
        else:
            # 任务执行成功的代码
            return "任务执行成功"
    except TaskFailed as e:
        # 捕获TaskFailed错误并返回错误信息给步骤函数
        raise e

{
  "Comment": "示例步骤函数",
  "StartAt": "LambdaFunction",
  "States": {
    "LambdaFunction": {
      "Type": "Task",
      "Resource": "arn:aws:lambda:region:account:function:functionName",
      "End": true,
      "Catch": [
        {
          "ErrorEquals": ["TaskFailed"],
          "Next": "HandleTaskFailed"
        }
      ]
    },
    "HandleTaskFailed": {
      "Type": "Pass",
      "Result": "任务执行失败",
      "End": true
    }
  }
}