19 lines
505 B
Python
19 lines
505 B
Python
import sys
|
|
import re
|
|
import zipfile
|
|
import glob
|
|
|
|
if len(sys.argv) != 2:
|
|
print('Usage: {} [wheel]'.format(sys.argv[0]))
|
|
sys.exit(1)
|
|
|
|
vcomp140_path = 'C:\\Windows\\System32\\vcomp140.dll'
|
|
|
|
for wheel_path in sorted(glob.glob(sys.argv[1])):
|
|
m = re.search(r'xgboost-(.*)-py3', wheel_path)
|
|
assert m, f'wheel_path = {wheel_path}'
|
|
version = m.group(1)
|
|
|
|
with zipfile.ZipFile(wheel_path, 'a') as f:
|
|
f.write(vcomp140_path, 'xgboost-{}.data/data/xgboost/vcomp140.dll'.format(version))
|