在 AWS Lambda 函数中调用端点时出现超时错误

Question

我正在尝试为我的项目创建一个 AWS Lambda 函数，它应该调用端点并返回模型的预测作为结果。

我将附加 IAM 角色的 sagemaker 调用端点 IAM 策略附加到 Lambda 函数。尽管如此，我在调用端点时遇到任务超时错误。

我在笔记本实例中运行 lambda 函数并能够在那里获得预测，但在 Lambda 函数中没有得到输出。

import os
import boto3
import json
import csv
# grab environment variables
ENDPOINT_NAME = os.environ['ENDPOINT_NAME']
runtime= boto3.client('runtime.sagemaker')


def lambda_handler(event,context):
    #print("Received event: " + json.dumps(event, indent=2))
    
    data = json.loads(json.dumps(event))
    payload = data['data']
    #print("payload:"+payload)
    #print("ENDPOINT NAME :"+ENDPOINT_NAME)
    response = runtime.invoke_endpoint(EndpointName=ENDPOINT_NAME,
                                       ContentType='text/csv',
                                       Body=payload)
    print(response)
    result = json.loads(response['Body'].read().decode())
    
    return result



{
  "errorMessage": "2023-03-04T00:49:45.554Z 38fee791-8b05-49e2-ab69-3035af43e499 Task timed out after 90.09 seconds"
}

请向我建议如何继续解决此错误。