Testing train/valid/test set build.