68ddd94e9f80f43cb623e41056c0759ee3883df0,pliers/extractors/text.py,BertLMExtractor,_mask,#BertLMExtractor#Any#Any#,663

Before Change

            self.mask_token, self.mask_pos = (mask, w_idx)
        else:
            raise ValueError("mask argument must be an integer or a string")
        nr_masks = len(np.where(np.array(mwds)=="[MASK]")[0])
        if nr_masks == 0:
            raise ValueError("No valid mask tokens found.")
        elif nr_masks > 1:
            raise ValueError("Too many masked items.")
        return mwds

    def _postprocess(self, preds, tok, wds, ons, dur):
        preds = preds[0].numpy()[:,1:-1,:]

After Change

        self.return_true = return_true

    def _mask(self, wds, mask):
        if not type(mask) in [int, str]:
            raise ValueError("mask argument must be an integer or a string")
        mwds = wds.copy()
        self.mask_token = mask if type(mask) == str else mwds[mask]
        self.mask_pos = np.where(np.array(mwds)==self.mask_token)[0][0]
        mwds[self.mask_pos] = "[MASK]"
        return mwds

    def _extract(self, stims, mask):
        """

Instances