modify dcu_version

openfold/__init__.py

@@ -2,5 +2,6 @@ from . import model
 from . import utils
 from . import np
 from . import resources
+from .version import __version__, __dcu_version__
 
 __all__ = ["model", "utils", "np", "data", "resources"]

setup.py

@@ -62,7 +62,7 @@ def get_abi():
     
 def get_version_add(sha: Optional[str] = None) -> str:
     openfold_root = os.path.dirname(os.path.abspath(__file__))
-    add_version_path = "version.py"
+    add_version_path = os.path.join(os.path.join(openfold_root, "openfold"), "version.py")
     if sha != 'Unknown':
         if sha is None:
             sha = get_sha(openfold_root)
@@ -71,7 +71,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
     # abi version
     version += "." + get_abi()
 
-     # dtk version
+    # dtk version
     if os.getenv("ROCM_PATH"):
         rocm_path = os.getenv('ROCM_PATH', "")
         rocm_version_path = os.path.join(rocm_path, '.info', "rocm_version")
@@ -84,7 +84,6 @@ def get_version_add(sha: Optional[str] = None) -> str:
     version += ".torch" + torch.__version__[:4]
 
     with open(add_version_path, encoding="utf-8", mode="w") as file:
-        # file.write("__version__='1.0.1'+'+{}'\n".format(version))
         file.write("__version__='1.0.1'\n")
         file.write("__dcu_version__='1.0.1+{}'\n".format(version))
     file.close()