diff --git a/gnes/service/preprocessor.py b/gnes/service/preprocessor.py index 50f2ecce..fb98c294 100644 --- a/gnes/service/preprocessor.py +++ b/gnes/service/preprocessor.py @@ -27,15 +27,15 @@ def post_init(self): self._model = self.load_model(BasePreprocessor) @handler.register(gnes_pb2.Request.TrainRequest) - def _handler_train_index(self, msg: 'gnes_pb2.Message'): + def _handler_train(self, msg: 'gnes_pb2.Message'): for d in msg.request.train.docs: self._model.apply(d) @handler.register(gnes_pb2.Request.IndexRequest) - def _handler_train_index(self, msg: 'gnes_pb2.Message'): + def _handler_index(self, msg: 'gnes_pb2.Message'): for d in msg.request.index.docs: self._model.apply(d) @handler.register(gnes_pb2.Request.QueryRequest) - def _handler_train_index(self, msg: 'gnes_pb2.Message'): + def _handler_query(self, msg: 'gnes_pb2.Message'): self._model.apply(msg.request.search.query)