File size: 797 Bytes
7e6c03a | 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 | import re
import os
files = [
"task_conditional_count.py",
"task_conditional_duration.py",
"task_between_events.py",
"task_event_density.py",
"task_duration_gap.py",
"task_temporal_arithmetic.py",
"task_temporal_loudness.py",
"task_multi_hop.py"
]
for filename in files:
with open(filename, 'r') as f:
content = f.read()
# Remove random.choice() around self.task_config["mcq_questions"]...
content = re.sub(r'random\.choice\(\s*(self\.task_config\[(?:\"|\')mcq_questions(?:\"|\')\]\[[^\]]+\])\s*\)', r'\1', content)
content = re.sub(r'random\.choice\(\s*(self\.task_config\[(?:\"|\')open_text_questions(?:\"|\')\]\[[^\]]+\])\s*\)', r'\1', content)
with open(filename, 'w') as f:
f.write(content)
print("Fixed files")
|