import sys
from glob import glob

path = f'/data/cv_datasets/processed/{sys.argv[1]}'
# folders = glob(f'{path}/**/split_*.json', recursive=True)

splits = []
folders1 = glob(f'{path}/*/')
for f1 in folders1:
    folders2 = glob(f'{f1}/*/')
    for f2 in folders2:
        folders3 = glob(f'{f2}/*/')
        for f3 in folders3:
            split = glob(f'{f3}/split_*.json')
            splits.append(split[0])

with open('splits.txt', 'w') as file:
    for split in sorted(splits):
        file.write(f'{split}\n')