U
    T?h                     @   s>   d dl Z d dlZd dlZddlmZ dd Zedkr:e  dS )    N   )update_onnx_opsetc                  C   sx   t jtjt dtj dd} | jdt	dddd | jd	t
jd
d | jdt
jdd |  }t|j|j|j d S )N:a  
                                     Update the ONNX opset of the model.
                                     New opset must be later than the existing one.
                                     If not specified will update to opset 15.
                                     )descriptionz--opsetF   zONNX opset to update to.)typerequireddefaulthelpinput_modelz%Provide path to ONNX model to update.)r   r
   output_modelz,Provide path to write updated ONNX model to.)argparseArgumentParserospathbasename__file__update_onnx_opset_helper__name__add_argumentintpathlibPath
parse_argsr   r   Zopsetr   )parserargs r   U/var/www/html/venv/lib/python3.8/site-packages/onnxruntime/tools/update_onnx_opset.pyr      s    	r   __main__)r   r   r   Zonnx_model_utilsr   r   r   r   r   r   r   <module>   s   