def on_delete()

in deployment/solution-assistant/src/lambda_function.py [0:0]

• 15 lines of code
• 2 McCabe index (conditional complexity)

def on_delete(event, __):
    # remove sagemaker endpoints
    solution_prefix = event["ResourceProperties"]["SolutionPrefix"]
    endpoint_names = [
        "{}-rcf".format(solution_prefix),
        "{}-xgb".format(solution_prefix),
        "{}-xgb-smote".format(solution_prefix)
    ]
    for endpoint_name in endpoint_names:
        delete_sagemaker_model(endpoint_name)
        delete_sagemaker_endpoint_config(endpoint_name)
        delete_sagemaker_endpoint(endpoint_name)

    # delete buckets
    model_data_bucket = event["ResourceProperties"]["ModelDataBucketName"]
    output_bucket = event["ResourceProperties"]["OutputBucketName"]
    bucket_delete_retry(model_data_bucket)
    bucket_delete_retry(output_bucket)