fix(router): respect num_part when set

gnes-ai · Aug 27, 2019 · d02cd75 · d02cd75
1 parent 87e4bcf
commit d02cd75
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gnes/service/router.py b/gnes/service/router.py
@@ -31,7 +31,7 @@ def post_init(self):
         self._pending = defaultdict(list)  # type: Dict[str, List]
 
     def _is_msg_complete(self, msg: 'gnes_pb2.Message', num_req: int) -> bool:
-        return (num_req == msg.envelope.num_part[-1]) or \
+        return (self.args.num_part is None and num_req == msg.envelope.num_part[-1]) or \
                (num_req == self.args.num_part)
 
     @handler.register(NotImplementedError)