U
    rh                     @  sH   d dl mZ d dlZd dlmZmZmZ ddddZdddd	ZdS )
    )annotationsN)	NodeProtocheckerloadNone)returnc                  C  s>   t d} | jdt dd |  }t|j}t| d S )Nzcheck-modelmodel_pbrbtype)	argparseArgumentParseradd_argumentFileType
parse_argsr   r   r   check_model)parserargsmodel r   4/tmp/pip-unpacked-wheel-xnis5xre/onnx/bin/checker.pyr      s
    

r   c                  C  sJ   t d} | jdt dd |  }t }||j  t	
| d S )Nz
check-nodenode_pbr	   r
   )r   r   r   r   r   r   ZParseFromStringr   readr   
check_node)r   r   noder   r   r   r      s    
r   )	
__future__r   r   Zonnxr   r   r   r   r   r   r   r   r   <module>   s   	