def lambda_handler()

in next_steps/operations/ml_ops_ds_sdk/lambda/stepfunction-createdatasetimportjob.py [0:0]

• 20 lines of code
• 1 McCabe index (conditional complexity)

def lambda_handler(event, context):

    datasetArn = event['dataset_arn']
    bucket = event['bucket_name']
    filename = event['file_name']
    roleArn = event['role_arn']
    
    create_dataset_import_job_response = personalize.create_dataset_import_job(
        jobName = "stepfunction-dataset-import-job",
        datasetArn = datasetArn,
        dataSource = {
            "dataLocation": "s3://{}/{}".format(bucket, filename)
        },
        roleArn = roleArn
    )
    
    dataset_import_job_arn = create_dataset_import_job_response['datasetImportJobArn']
    print(json.dumps(create_dataset_import_job_response, indent=2))




    # TODO implement
    return {
        'statusCode': 200,
        'dataset_import_job_arn': dataset_import_job_arn,
        'datasetGroupArn': event['datasetGroupArn']
        #'body': json.dumps('Hello from Lambda!')
    }