in dags/inference/utils/trtllm_bench_jsonl_converter.py [0:0]
def extract_data(patterns, data):
extracted = {}
for section, section_patterns in patterns.items():
extracted[section] = {}
for field, pattern in section_patterns.items():
match = re.search(pattern, data)
if match:
extracted[section][field] = match.group(1)
return extracted