and if if args.training_set is not None we just take the labels given to us without modification?