o ©Ì³iÜã@sHddlmZddlZdedefdd„Zdd„Zd ed edefd d „ZdS) é)ÚdatetimeNÚversionÚreturncCs|tjvp tj|kS)ai Check if torch version is greater than or equal to the given version. Args: version (str): The torch version to compare against Returns: bool: True if torch version is greater than or equal to the given version. Example: >>> print(torch.__version__) 2.4.0 >>> torch_version_ge("2.0") True )ÚtorchÚ __version__)r©rúL/home/ubuntu/.local/lib/python3.10/site-packages/torchtune/utils/_version.pyÚtorch_version_ge sr cCsttjdƒ S)NÚ git_version)ÚhasattrrrrrrrÚ _is_fbcodesr Úao_version_strÚdatecCs0t | d¡d d¡dd¡}|t |d¡kS)z¶ Compare a torchao nightly version to a date of the form %Y-%m-%d. Returns True if the nightly version is greater than or equal to the date, False otherwise ú+rÚdevéz%Y%m%dz%Y-%m-%d)rÚstrptimeÚsplit)r rÚ ao_datetimerrrÚ_nightly_version_ge#sÿr)rrÚstrÚboolr r rrrrrÚs