Commit d5dcc5c6 authored by lijian6's avatar lijian6
Browse files

Modify setup.py


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent 02afa546
__dcu_version__ = '0.29.0+das1.2.gitfe0fecc.abi0.dtk2404'
__dcu_version__ = '0.29.0+das1.2.git02afa54.abi0.dtk2404'
......@@ -254,9 +254,9 @@ install_requires = [
version_range_max = max(sys.version_info[1], 10) + 1
def get_sha(ort_root: Union[str, Path]) -> str:
def get_sha(root_dir: Union[str, Path]) -> str:
try:
return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=ort_root).decode('ascii').strip()
return subprocess.check_output(['git', 'rev-parse', 'HEAD'], cwd=root_dir).decode('ascii').strip()
except Exception:
return 'Unknown'
......@@ -272,11 +272,11 @@ def get_abi():
def get_version_add(sha: Optional[str] = None) -> str:
version=''
ort_root = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(ort_root, "DIFFUSERS_VERSION_NUMBER")
root_dir = os.path.dirname(os.path.abspath(__file__))
add_version_path = os.path.join(root_dir, "DIFFUSERS_VERSION_NUMBER")
if sha != 'Unknown':
if sha is None:
sha = get_sha(ort_root)
sha = get_sha(root_dir)
version = 'git' + sha[:7]
# abi
......@@ -299,7 +299,7 @@ def get_version_add(sha: Optional[str] = None) -> str:
file.writelines(lines)
file.close()
init_path=os.path.join(ort_root, "src/diffusers/__init__.py")
init_path=os.path.join(root_dir, "src/diffusers/__init__.py")
with open(init_path, 'r',encoding='utf-8') as file:
lines = file.readlines()
lines[1] = "__dcu_version__ = '0.29.0+das1.2.{}'\n".format(version)
......@@ -310,8 +310,8 @@ def get_version_add(sha: Optional[str] = None) -> str:
def get_version():
get_version_add()
ort_root = os.path.dirname(os.path.abspath(__file__))
version_file = os.path.join(ort_root, "DIFFUSERS_VERSION_NUMBER")
root_dir = os.path.dirname(os.path.abspath(__file__))
version_file = os.path.join(root_dir, "DIFFUSERS_VERSION_NUMBER")
with open(version_file, encoding='utf-8') as f:
exec(compile(f.read(), version_file, 'exec'))
return locals()['__dcu_version__']
......
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