jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

fix hook

Browse Source
This commit is contained in:
Myhs-phz 2025-03-19 09:14:52 +00:00
parent ffe00a830d
commit b1c2f60d32
1 changed files with 20 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
tools/update_dataset_suffix.py
View File

@ -94,21 +94,21 @@ def check_and_rename(filepath):
return None, None
def update_imports(data):
python_file, name_pairs = data
for filepath, new_file in name_pairs:
old_name = os.path.basename(filepath)[:-3]
new_name = os.path.basename(new_file)[:-3]
if not os.path.exists(python_file):
return
with open(python_file, 'r') as file:
filedata = file.read()
# Replace the old name with new name
new_data = filedata.replace(old_name, new_name)
if filedata != new_data:
with open(python_file, 'w') as file:
file.write(new_data)
# print(f"Updated imports in {python_file}")
# def update_imports(data):
# python_file, name_pairs = data
# for filepath, new_file in name_pairs:
# old_name = os.path.basename(filepath)[:-3]
# new_name = os.path.basename(new_file)[:-3]
# if not os.path.exists(python_file):
# return
# with open(python_file, 'r') as file:
# filedata = file.read()
# # Replace the old name with new name
# new_data = filedata.replace(old_name, new_name)
# if filedata != new_data:
# with open(python_file, 'w') as file:
# file.write(new_data)
# # print(f"Updated imports in {python_file}")
def main():
@ -134,11 +134,11 @@ def main():
return
with Pool(16) as p:
p.starmap(os.rename, name_pairs)
root_folder = 'configs'
python_files = glob.glob(f'{root_folder}/**/*.py', recursive=True)
update_data = [(python_file, name_pairs) for python_file in python_files]
with Pool(16) as p:
p.map(update_imports, update_data)
# root_folder = 'configs'
# python_files = glob.glob(f'{root_folder}/**/*.py', recursive=True)
# update_data = [(python_file, name_pairs) for python_file in python_files]
# with Pool(16) as p:
# p.map(update_imports, update_data)
if __name__ == '__main__':