Try sending replies direct

Author: ianthomas23

ipykernel/kernelbase.py

@@ -425,7 +425,8 @@ async def dispatch_shell(self, msg, /, stream=None, subshell_id: str | None = No
             except Exception:
                 self.log.debug("Unable to signal in pre_handler_hook:", exc_info=True)
             try:
-                result = handler(stream, idents, msg)
+                #result = handler(stream, idents, msg)
+                result = handler(self.shell_stream, idents, msg)
                 if inspect.isawaitable(result):
                     await result
             except Exception: