如何正确编写返回numpy数组的sagemaker tensorflow input_handler()？

import numpy as np
import json

def input_handler(event, context):
    # 解析输入数据
    input_data = json.loads(event['body'])
    
    # 处理输入数据，生成NumPy数组
    # ...
    
    # 返回NumPy数组
    return {
        'statusCode': 200,
        'body': json.dumps(input_data.tolist())
    }

def input_handler(event, context):
    # 解析输入数据
    input_data = json.loads(event['body'])
    
    # 处理输入数据，生成NumPy数组
    features = []
    labels = []
    for sample in input_data:
        features.append(sample['features'])
        labels.append(sample['label'])
    features = np.array(features)
    labels = np.array(labels)
    
    # 返回NumPy数组
    return {
        'statusCode': 200,
        'body': json.dumps({
            'features': features.tolist(),
            'labels': labels.tolist()
        })
    }

[
    {
        "features": [1, 2, 3],
        "label": 0
    },
    {
        "features": [4, 5, 6],
        "label": 1
    },
    ...
]