def setup_pretrained_endpoints()

in credit-card-fraud-detection/neptune_ml_utils.py [0:0]

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

def setup_pretrained_endpoints(s3_bucket_uri: str, setup_node_classification: bool,
                               setup_node_regression: bool, setup_link_prediction: bool, \
                               setup_edge_classification: bool, setup_edge_regression: bool):
    delete_pretrained_data(setup_node_classification,
                           setup_node_regression, setup_link_prediction,
                           setup_edge_classification, setup_edge_regression)
    try:
        return PretrainedModels().setup_pretrained_endpoints(s3_bucket_uri, setup_node_classification,
                                                             setup_node_regression, setup_link_prediction,
                                                             setup_edge_classification, setup_edge_regression)
    except Exception as e:
        logging.error(e)