def lambda_handler()

in ml_ops/visualization_blog/lambdas/deletedatasetimport/deletedataset.py [0:0]


def lambda_handler(event, context):
    try:
        LOADER.forecast_cli.delete_dataset_import_job(
            DatasetImportJobArn=event['DatasetImportJobArn']
        )
        actions.take_action_delete(
            LOADER.forecast_cli.describe_dataset_import_job(
                DatasetImportJobArn=event['DatasetImportJobArn']
            )['Status']
        )

    except (LOADER.forecast_cli.exceptions.ResourceNotFoundException, KeyError):
        LOADER.logger.info('Import job not found! Passing.')

    return event