20 lines
528 B
Python
20 lines
528 B
Python
import sys
|
||
import ruamel.yaml
|
||
|
||
yaml = ruamel.yaml.YAML()
|
||
yaml.preserve_quotes = True
|
||
yaml.default_flow_style = False
|
||
file_path = 'metafile.yml'
|
||
# 读取YAML文件内容
|
||
with open(file_path, 'r') as file:
|
||
data = yaml.load(file)
|
||
# 遍历模型列表
|
||
for model in data.get('Models', []):
|
||
# 为每个模型添加Weights键值对，确保名称被正确引用
|
||
model['Weights'] = model['Name']
|
||
|
||
# 将修改后的数据写回文件
|
||
with open(file_path, 'w') as file:
|
||
yaml.dump(data, file)
|
||
|
||
print("Modifications saved to the file.") |