fix(preprocessor): fix preprocessor service handler function name error

gnes-ai · Jul 17, 2019 · 3836020 · 3836020
1 parent 1229f5c
commit 3836020
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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)