fix(encoder): fix bug for encoder service dealing with empty doc

gnes-ai · Jul 26, 2019 · 1dff06f · 1dff06f
1 parent 7e43d5a
commit 1dff06f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gnes/service/encoder.py b/gnes/service/encoder.py
@@ -41,8 +41,9 @@ def _handler_index(self, msg: 'gnes_pb2.Message'):
         vecs = self._model.encode(chunks)
         s = 0
         for d in msg.request.index.docs:
-            d.chunk_embeddings.CopyFrom(array2blob(vecs[s:(s + len(d.chunks))]))
+            if len(d.chunks):
-            s += len(d.chunks)
+                d.chunk_embeddings.CopyFrom(array2blob(vecs[s:(s + len(d.chunks))]))
+                s += len(d.chunks)
 
     @handler.register(gnes_pb2.Request.TrainRequest)
     def _handler_train(self, msg: 'gnes_pb2.Message'):