Change "train_only" task name to "train" to match documentation.