5275d70dcc655a3a99333c0361dc8c24c77ed2c5,pliers/extractors/text.py,BertLMExtractor,_postprocess,#BertLMExtractor#Any#Any#Any#Any#Any#Any#,714

Before Change

            data += [stims.name]
            feat += ["sequence"]
        if len(self.target) > 1:
            self.target = [self.target]
        ons, dur = map(lambda x: listify(x[self.mask_pos]), [ons, dur])
        return data, feat, ons, dur

    def _return_masked_word(self, preds, feat, data):

After Change

            sub_idx = out_idx
        out_idx = [idx for idx in out_idx if idx in sub_idx]
        feat = self.tokenizer.convert_ids_to_tokens(out_idx)
        feat = [f.upper() for f in feat]
        data = [listify(p) for p in preds[0,self.mask_pos,out_idx]]
        if self.return_masked_word:
            feat, data = self._return_masked_word(preds, feat, data)

Instances