def lambda_handler()

in automation/ml_ops/lambdas/list-datasets/list-datasets.py [0:0]


def lambda_handler(event, context):
    try:
        response = LOADER.personalize_cli.list_datasets(
            datasetGroupArn=event['datasetGroupArn'],
            maxResults = 100
        )

        return json.loads(json.dumps(response['datasets'], default=str))
    except Exception as e:
        LOADER.logger.error(f'Error listing datasets {e}')
        raise e