U
    q±Ëhx  ã                   @  s:   d dl mZ d dlZd dlm  mZ dddœdd„ZdS )é    )ÚannotationsNz6onnx.ModelProto | onnx.FunctionProto | onnx.GraphProtoÚstr)ÚprotoÚreturnc                 C  sZ   t | tjƒrt |  ¡ ¡S t | tjƒr4t |  ¡ ¡S t | tjƒrNt 	|  ¡ ¡S t
dƒ‚d S )NzUnsupported argument type.)Ú
isinstanceÚonnxZ
ModelProtoÚCZmodel_to_textZSerializeToStringZFunctionProtoZfunction_to_textZ
GraphProtoZgraph_to_textÚ	TypeError)r   © r
   ú0/tmp/pip-unpacked-wheel-xnis5xre/onnx/printer.pyÚto_text
   s    r   )Ú
__future__r   r   Zonnx.onnx_cpp2py_export.printerZonnx_cpp2py_exportÚprinterr   r   r
   r
   r
   r   Ú<module>   s   