Error in create_train_test_dirs()

are you going to solve the problem or just tell us to do what we already do?