    def tokens(self) -> Set[int]:
        """Returns the set of tokens in constraints."""
        return set(self.token_counts().keys())
Returns the set of tokens in constraints.
fairseq/token_generation_constraints.py