examples/research_projects/bertabs/modeling_bertabs.py/PenaltyBuilder/length_penalty
class PenaltyBuilder:    def length_penalty(self):
        if self.length_pen == "wu":
            return self.length_wu
        elif self.length_pen == "avg":
            return self.length_average
        else:
            return self.length_none