o he@stddlZddlmZddlmZmZddlmZeej dZ dee efde d e fd d Z de d e fd dZdS)N)Union)Versionparse)STR_OPERATION_TO_FUNCtorchlibrary_or_version operationrequirement_versioncCsX|tvrtdttd|t|}t|tr%ttj |}||t|S)a Compares a library version to some requirement using a given operation. Args: library_or_version (`str` or `packaging.version.Version`): A library name or a version to check. operation (`str`): A string representation of an operator, such as `">"` or `"<="`. requirement_version (`str`): The version to compare the library version against z`operation` must be one of z , received ) rkeys ValueErrorlist isinstancestrr importlibmetadataversion)rr r rO/var/www/html/ai/venv/lib/python3.10/site-packages/accelerate/utils/versions.pycompare_versionss  rrcCs tt||S)a  Compares the current PyTorch version to a given reference with an operation. Args: operation (`str`): A string representation of an operator, such as `">"` or `"<="` version (`str`): A string version of PyTorch )r torch_version)r rrrris_torch_version.s r)importlib.metadatartypingrpackaging.versionrr constantsrrrrrrrrrrrs