File size: 815 Bytes
e8ea4f4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
#!/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")