Commit 52919c63 authored by zhuww's avatar zhuww
Browse files

modify version

parent edcb2f85
...@@ -106,7 +106,7 @@ def get_sha(pytorch_root: Union[str, Path]) -> str: ...@@ -106,7 +106,7 @@ def get_sha(pytorch_root: Union[str, Path]) -> str:
def get_version_add(sha: Optional[str] = None) -> str: def get_version_add(sha: Optional[str] = None) -> str:
add_version_path = "jax/version.py" add_version_path = "version.py"
if sha != 'Unknown': if sha != 'Unknown':
if sha is None: if sha is None:
sha_path = os.getenv('FASTFOLD_DOWNLOAD_PATH', "") sha_path = os.getenv('FASTFOLD_DOWNLOAD_PATH', "")
...@@ -118,8 +118,7 @@ def get_version_add(sha: Optional[str] = None) -> str: ...@@ -118,8 +118,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
version += "." + version_dtk version += "." + version_dtk
with open(add_version_path, encoding="utf-8", mode="w") as file: with open(add_version_path, encoding="utf-8", mode="w") as file:
file.write("__version__='0.2.21'+'+{}'\n".format(version)) file.write("__version__='0.2.0'+'+{}'\n".format(version))
file.write("_minimum_jaxlib_version='0.1.69'\n")
file.close() file.close()
...@@ -145,5 +144,5 @@ setup( ...@@ -145,5 +144,5 @@ setup(
ext_modules=ext_modules, ext_modules=ext_modules,
package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*']}, package_data={'fastfold': ['model/fastnn/kernel/cuda_native/csrc/*']},
cmdclass={'build_ext': BuildExtension} if ext_modules else {}, cmdclass={'build_ext': BuildExtension} if ext_modules else {},
#install_requires=['einops', 'colossalai'], install_requires=['einops', 'colossalai'],
) )
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment