a :gh@sUddlmZddlZddlZddlZddlZdZdZded<dZ ded<ddd d Z ddd d d Z ddd ddZ ddddZ dddddZddZe ZdZddZeeZeeZdS)) annotationsNz0.4.30str_release_versionz str | None _git_hash)returncCstdur tSttpttSN)r_version_from_git_tree_version_version_from_todays_dater r [/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/jax/version.py_get_version_string!sr ) base_versionrcCstjd}|d|S)N%Y%m%d.dev)datetimedatetodaystrftime)r datestringr r r r )sr cCsztjtjt}tjgd|tjtjd}|\}}| dd\}}t j t|d}|svJ|sJWnYdS0|d|d|SdS)N)gitshowz-sz--format=%at-%hHEAD)cwdstdoutstderr-rr+)ospathdirnamerealpath__file__ subprocessPopenPIPE communicatedecodestripsplitrr fromtimestampintr isnumericisalnum)rZroot_directorypr_ timestampZ commit_hashrr r r r.s   rcCsXtdur tStjds$tjdr,ttStjdsDtjdrHtSttpVttS)a]Determine the version at build time. The returned version string depends on which environment variables are set: - if JAX_RELEASE or JAXLIB_RELEASE are set: version looks like "0.4.16" - if JAX_NIGHTLY or JAXLIB_NIGHTLY are set: version looks like "0.4.16.dev20230906" - if none are set: version looks like "0.4.16.dev20230906+ge58560fdc NZ JAX_NIGHTLYZJAXLIB_NIGHTLYZ JAX_RELEASEZJAXLIB_RELEASE)rrenvirongetr r rr r r r _get_version_for_buildBsr4None)fnamerc Cst}d}d|}t|}|}||dkrHtd|d||||}tj d}|rd}d|}||dkrtd|d||||}| |d S) zJUsed by setup.py to write the specified version info into the source tree.z _release_version: str = '0.4.30'z_release_version: str = zBuild: could not find z in Z JAX_GIT_HASHz_git_hash: str | None = Nonez_git_hash: str = N) r4pathlibPath read_textcount RuntimeErrorreplacerr2r3 write_text) r6Zrelease_versionZold_version_stringZnew_version_stringZfhandlecontentsZgithashZold_githash_stringZnew_githash_stringr r r _write_versionSs       r@csLddlm}ddlm}Gfddd|}Gfddd|}t||dS) Nr)build_py)sdistcseZdZfddZZS)z _get_cmdclass.._build_pycsVtdur8tj|jtjt}tj|r8t|t tdurRt |dSr) rrr join build_libbasenamer#isfileunlinksuperrunr@)selfZthis_file_in_build_dir __class__pkg_source_pathr r rIns     z$_get_cmdclass.._build_py.run)__name__ __module__ __qualname__rI __classcell__r rMrLr _build_pymsrTcseZdZfddZZS)z_get_cmdclass.._sdistcs6t||tdur2ttj|tjtdSr) rHmake_release_treerr@rr rCrEr#)rJbase_dirfilesrKr r rU~s   z/_get_cmdclass.._sdist.make_release_tree)rNrOrPrUrQr rRrSr _sdist}srX)rBrA)Zsetuptools.command.build_pyrAZsetuptools.command.sdistrBdict)rMZ build_py_origZ sdist_origrTrXr rRr _get_cmdclassis   rZz0.4.27cCstdd|dDS)Ncss|]}|rt|VqdSr)isdigitr,).0ir r r z$_version_as_tuple...)tupler*) version_strr r r _version_as_tuplesrc) __future__rrrr8r$r r__annotations__rr r rr4r@rZ __version__Z_minimum_jaxlib_versionrc__version_info__Z_minimum_jaxlib_version_infor r r r s$