use get_shell_msg helper method (#368)

* use get_shell_msg helper method
jupyter-server · Aug 24, 2021 · 85a4899 · 85a4899
1 parent 213a817
commit 85a4899
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/kernel_gateway/services/kernels/manager.py b/kernel_gateway/services/kernels/manager.py
@@ -101,7 +101,7 @@ def start_kernel(self, *args, **kwargs):
                     # executed
                     if self.parent.personality.should_seed_cell(code):
                         client.execute(code)
-                        msg = client.shell_channel.get_msg(block=True)
+                        msg = client.get_shell_msg()
                         if msg['content']['status'] != 'ok':
                             # Shutdown the channels to remove any lingering ZMQ messages
                             client.stop_channels()