I am not sure this is right... unless self.args.trainer_args defaults to None I could see this becoming an issue. Will investigate