in lambda/stepfunctioncreatedataset.py [0:0]
def lambda_handler(event, context):
# TODO implement
dataset_type = "INTERACTIONS"
datasetGroupArn = event['datasetGroupArn']
create_dataset_response = personalize.create_dataset(
name = "personalize-stepfunction-dataset",
datasetType = dataset_type,
datasetGroupArn = event['datasetGroupArn'],
schemaArn = event['schemaArn']
)
dataset_arn = create_dataset_response['datasetArn']
print(json.dumps(create_dataset_response, indent=2))
return {
'statusCode': 200,
'dataset_arn': dataset_arn,
'datasetGroupArn': datasetGroupArn
#'body': json.dumps('Hello from Lambda!')
}