def extract_data()

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