This dataset comes from the automatic collection in simplescaling/s1's data/collect_data.py, It may be necessary to replace 's1/s1K' with 'simplescaling/s1K' in the file.
You don't need to run again: data/add_aime.py. If you run it again, it will cause duplicate data.