def find_keys(args):
    filenames = []
    for indir in args.indir:
        task = next(indir.iterdir())
        for method in task.iterdir():
            seed = next(indir.iterdir())
            filenames += list(seed.glob('**/*.jsonl'))
    keys = set()
    for filename in filenames:
        keys |= set(load_jsonl(filename).columns)
    print(f'Keys      ({len(keys)}):', ', '.join(keys), flush=True)
def find_keys(args):
    filenames = []
    for indir in args.indir:
        task = next(indir.iterdir())
        for method in task.iterdir():
            seed = next(indir.iterdir())
            filenames += list(seed.glob('**/*.jsonl'))
    keys = set()
    for filename in filenames:
        keys |= set(load_jsonl(filename).columns)
    print(f'Keys      ({len(keys)}):', ', '.join(keys), flush=True)