def vectorize(self, *args, **kwargs): """ Overrides BiencoderAgent to not add start/end. """ kwargs['add_start'] = False kwargs['add_end'] = False return TorchRankerAgent.vectorize(self, *args, **kwargs)
def vectorize(self, *args, **kwargs): """ Add the start and end token to the text. """ kwargs['add_start'] = True kwargs['add_end'] = True obs = TorchRankerAgent.vectorize(self, *args, **kwargs) return obs