def dump_ffn_input_completions()

in src/scripts/extract_data.py [0:0]

• 5 lines of code
• 3 McCabe index (conditional complexity)

def dump_ffn_input_completions(entity_name, dataset_name, data, keys):
    entity_folder = f"./data/{dataset_name}/{entity_name}"
    os.makedirs(entity_folder, exist_ok=True)
    with open(f"{entity_folder}/completions.json", "w") as f:
        json.dump({k: [item[entity_name] for item in data[k]] for k in keys}, f, indent=2)