def get_operators()

in scripts/collect_models.py [0:0]

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

def get_operators(model: onnx.ModelProto) -> Set[str]:
    """
    Recursively traverses the ONNX graph and returns a set of operator names.

    Args:
        model: Loaded ONNX model.

    Returns:
        Set of operator names.
    """
    operators: Set[str] = set()

    def traverse_graph(graph: onnx.GraphProto):
        for node in graph.node:
            operators.add(node.op_type)
            for attr in node.attribute:
                if attr.type == onnx.AttributeProto.GRAPH:
                    traverse_graph(attr.g)

    traverse_graph(model.graph)
    return operators