)tieddlZddlmZddlmZmZddlmZeejjdZ dee efde d e fd Z de d e fd Zy) N)Union)Versionparse)STR_OPERATION_TO_FUNCtorchlibrary_or_version operationrequirement_versionc,|tjvr,tdttjd|t|}t |t r(t tjj|}||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)r r r s h/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/accelerate/utils/versions.pycompare_versionsrs-22446tP#QR '/B)C DDrc$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 rs ris_torch_versionr.s M9g >>r)importlib.metadatartypingrpackaging.versionrr constantsrrrrrrrrrr sk,,i((009: EsG|)<EEcfE( ? ?c ?r