Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.11/site-packages/onnx/parser.py: 41%

1# Copyright (c) ONNX Project Contributors 

2# 

3# SPDX-License-Identifier: Apache-2.0 

4from __future__ import annotations 

5 

6import onnx 

7import onnx.onnx_cpp2py_export.parser as C # noqa: N812 

8 

9 

10class ParseError(Exception): 

11 pass 

12 

13 

14def parse_model(model_text: str) -> onnx.ModelProto: 

15 """Parse a string to build a ModelProto. 

16 

17 Arguments: 

18 model_text (string): formatted string 

19 Returns: 

20 ModelProto 

21 """ 

22 (success, msg, model_proto_str) = C.parse_model(model_text) 

23 if success: 

24 return onnx.load_from_string(model_proto_str) 

25 raise ParseError(msg) 

26 

27 

28def parse_graph(graph_text: str) -> onnx.GraphProto: 

29 """Parse a string to build a GraphProto. 

30 

31 Arguments: 

32 graph_text (string): formatted string 

33 Returns: 

34 GraphProto 

35 """ 

36 (success, msg, graph_proto_str) = C.parse_graph(graph_text) 

37 if success: 

38 graph_proto = onnx.GraphProto() 

39 graph_proto.ParseFromString(graph_proto_str) 

40 return graph_proto 

41 raise ParseError(msg) 

42 

43 

44def parse_function(function_text: str) -> onnx.FunctionProto: 

45 """Parse a string to build a FunctionProto. 

46 

47 Arguments: 

48 function_text (string): formatted string 

49 Returns: 

50 FunctionProto 

51 """ 

52 (success, msg, function_proto_str) = C.parse_function(function_text) 

53 if success: 

54 function_proto = onnx.FunctionProto() 

55 function_proto.ParseFromString(function_proto_str) 

56 return function_proto 

57 raise ParseError(msg) 

58 

59 

60def parse_node(node_text: str) -> onnx.NodeProto: 

61 """Parse a string to build a NodeProto. 

62 

63 Arguments: 

64 node_text: formatted string 

65 Returns: 

66 NodeProto 

67 """ 

68 (success, msg, node_proto_str) = C.parse_node(node_text) 

69 if success: 

70 node_proto = onnx.NodeProto() 

71 node_proto.ParseFromString(node_proto_str) 

72 return node_proto 

73 raise ParseError(msg)