61141608c3529645ef0895d38c72c0346797d692,pliers/extractors/text.py,BertLMExtractor,_postprocess,#BertLMExtractor#Any#Any#Any#Any#Any#Any#,717

Before Change

        if self.return_input:
            data += [stims.name]
            feat += ["sequence"]
        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):
        if self.mask_token in self.tokenizer.vocab:

After Change

        if self.return_input:
            data += [stims.name]
            feat += ["sequence"]
        mask_ons = listify(stims.elements[self.mask_pos].onset)
        mask_dur = listify(stims.elements[self.mask_pos].duration)
        return data, feat, mask_ons, mask_dur
    

Instances