o
    åeÀ  ã                   @   sV   d dl Z d dlm  mZ G dd„ deƒZdede jfdd„Z	dede j
fd	d
„ZdS )é    Nc                   @   s   e Zd ZdS )Ú
ParseErrorN)Ú__name__Ú
__module__Ú__qualname__© r   r   ú;D:\Projects\ConvertPro\env\Lib\site-packages\onnx/parser.pyr      s    r   Ú
model_textÚreturnc                 C   s&   t  | ¡\}}}|rt |¡S t|ƒ‚)zˆParse a string to build a ModelProto.

    Arguments:
        model_text (string): formatted string
    Returns:
        ModelProto
    )ÚCÚparse_modelÚonnxZload_from_stringr   )r   ÚsuccessÚmsgZmodel_proto_strr   r   r   r   
   s   
r   Ú
graph_textc                 C   s2   t  | ¡\}}}|rt ¡ }| |¡ |S t|ƒ‚)zˆParse a string to build a GraphProto.

    Arguments:
        graph_text (string): formatted string
    Returns:
        GraphProto
    )r
   Úparse_graphr   Ú
GraphProtoZParseFromStringr   )r   r   r   Zgraph_proto_strÚGr   r   r   r      s   
r   )r   Zonnx.onnx_cpp2py_export.parserZonnx_cpp2py_exportÚparserr
   Ú	Exceptionr   ÚstrZ
ModelProtor   r   r   r   r   r   r   Ú<module>   s
   