(self, tokenizer, type_schema, *args, **kwargs)
| 252 | # Src -> ((ET)(ET)), Text2ET 使用 |
| 253 | class ETConstraintDecoder(ConstraintDecoder): |
| 254 | def __init__(self, tokenizer, type_schema, *args, **kwargs): |
| 255 | super().__init__(tokenizer, *args, **kwargs) |
| 256 | self.tree_end = '<tree-end>' |
| 257 | self.type_tree = get_label_name_tree(type_schema.type_list, |
| 258 | tokenizer=self.tokenizer, |
| 259 | end_symbol=self.tree_end) |
| 260 | self.type_start = self.tokenizer.convert_tokens_to_ids([type_start])[0] |
| 261 | self.type_end = self.tokenizer.convert_tokens_to_ids([type_end])[0] |
| 262 | |
| 263 | def check_state(self, tgt_generated): |
| 264 | if tgt_generated[-1] == self.tokenizer.pad_token_id: # t5-base |
nothing calls this directly
no test coverage detected