o
    toi                     @  s@   d dl mZ d dlZd dlmZmZmZ d
ddZd
dd	ZdS )    )annotationsN)	NodeProtocheckerloadreturnNonec                  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   D/home/ubuntu/.local/lib/python3.10/site-packages/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   ParseFromStringr   readr   
check_node)r   r   noder   r   r   r      s   
r   )r   r   )	
__future__r   r   onnxr   r   r   r   r   r   r   r   r   <module>   s
   
	