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
