def delete_pretrained_endpoints()

in src/graph_notebook/notebooks/03-Neptune-ML/03-Sample-Applications/04-Telco-Networks/neptune_ml_utils.py [0:0]

• 21 lines of code
• 12 McCabe index (conditional complexity)

def delete_pretrained_endpoints(endpoints: dict):
    sm = boto3.client("sagemaker")
    try:
        if 'node_classification_endpoint_name' in endpoints and endpoints['node_classification_endpoint_name']:
            sm.delete_endpoint(
                EndpointName=endpoints['node_classification_endpoint_name']["EndpointName"])
        if 'node_regression_endpoint_name' in endpoints and endpoints['node_regression_endpoint_name']:
            sm.delete_endpoint(
                EndpointName=endpoints['node_regression_endpoint_name']["EndpointName"])
        if 'prediction_endpoint_name' in endpoints and endpoints['prediction_endpoint_name']:
            sm.delete_endpoint(
                EndpointName=endpoints['prediction_endpoint_name']["EndpointName"])
        if 'edge_classification_endpoint_name' in endpoints and endpoints['edge_classification_endpoint_name']:
            sm.delete_endpoint(
                EndpointName=endpoints['edge_classification_endpoint_name']["EndpointName"])
        if 'edge_regression_endpoint_name' in endpoints and endpoints['edge_regression_endpoint_name']:
            sm.delete_endpoint(
                EndpointName=endpoints['edge_regression_endpoint_name']["EndpointName"])
        print(f'Endpoint(s) have been deleted')
    except Exception as e:
        logging.error(e)