o Xi @sxdZddlmZddlZddlmZmZddlZdd d Z dd d Z dddZ dddZ dddZ ddZdddZdS)zVersion utils for testing.) annotationsN)CallableSequenceversionstrreturnboolcC&ddl}tj|jjtj|jkS)zAReturns True if the ONNX version is older than the given version.rN)onnx packagingrparse __version__release)rr rV/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/_internal/version_utils.pyonnx_older_than  rcCr )zBReturns True if the torch version is older than the given version.rN)torchr rr r r)rrrrrtorch_older_thanrr bool | NonecCs@zddl}Wn tyYdSwtj|jjtj|jkS)zIReturns True if the transformers version is older than the given version.rN) transformers ImportErrorr rr r r)rrrrrtransformers_older_than!s   rcCr )zHReturns True if the onnxruntime version is older than the given version.rN) onnxruntimer rr r r)rrrrronnxruntime_older_than.rrcCr )zBReturns True if the numpy version is older than the given version.rN)numpyr rr r r)rrrrrnumpy_older_than8rrcCs,z ddl}|s JWdStyYdSw)z#Tells if transformers is installed.rNTF)rr)rrrrhas_transformersBs rwarnsWarning | Sequence[Warning]rcsfdd}|S)zlCatches warnings. Args: warns: warnings to ignore Returns: decorated function cs*dur tddfdd}|S)Nzwarns cannot be None for 'z'.cs@ttd|WdS1swYdS)Nignore)warningscatch_warnings simplefilter)self)fctrrrcall_f[s  $z0ignore_warnings..wrapper..call_f)AssertionError)r%r&r)r%rwrapperWsz ignore_warnings..wrapperr)rr)rr(rignore_warningsMs  r*)rrrr)rrrr)rrrr)__doc__ __future__rr!typingrrpackaging.versionr rrrrrrr*rrrrs