qlib / fix_qlib_version.py
guohanghui's picture
Create fix_qlib_version.py
e8ea4f4 verified
Raw
History Blame Contribute Delete
815 Bytes
#!/usr/bin/env python3
import os
file_path = '/app/qlib/source/qlib/__init__.py'
# 读取文件
with open(file_path, 'r') as f:
lines = f.readlines()
# 修改内容
new_lines = []
skip_next = False
for i, line in enumerate(lines):
if 'from setuptools_scm import get_version' in line:
new_lines.append('try:\n')
new_lines.append(' from setuptools_scm import get_version\n')
skip_next = True
elif skip_next and '__version__ = get_version' in line:
new_lines.append(' ' + line)
new_lines.append('except LookupError:\n')
new_lines.append(' __version__ = "0.9.0"\n')
skip_next = False
else:
new_lines.append(line)
# 写回文件
with open(file_path, 'w') as f:
f.writelines(new_lines)
print("Fixed qlib version detection")