def get_edgelists()

in source/sagemaker/sagemaker_graph_fraud_detection/dgl_fraud_detection/graph.py [0:0]

• 6 lines of code
• 4 McCabe index (conditional complexity)

def get_edgelists(edgelist_expression, directory):
    if "," in edgelist_expression:
        return edgelist_expression.split(",")
    files = os.listdir(directory)
    compiled_expression = re.compile(edgelist_expression)
    return [filename for filename in files if compiled_expression.match(filename)]