feat(encoder): add debug hook

gnes-ai · Sep 6, 2019 · 105a0ab · 105a0ab
1 parent e285fa4
commit 105a0ab
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gnes/service/base.py b/gnes/service/base.py
@@ -398,8 +398,8 @@ def _run(self, ctx):
 
                         # choose output sock
                         if msg.request and msg.request.WhichOneof('body') and \
-                                type(getattr(msg.request,
-                                             msg.request.WhichOneof('body'))) == gnes_pb2.Request.ControlRequest:
+                                isinstance(getattr(msg.request, msg.request.WhichOneof('body')),
+                                           gnes_pb2.Request.ControlRequest):
                             o_sock = ctrl_sock
                         else:
                             o_sock = out_sock