def restart_kernel(self, now=False, **kwds): self.shutdown_kernel() self.start_kernel(**kwds) @property def has_kernel(self): return self.kernel is not None def _kill_kernel(self): self.kernel = None def interrupt_kernel(self): raise NotImplementedError("Cannot interrupt in-process kernel.") def signal_kernel(self, signum): raise NotImplementedError("Cannot signal in-process kernel.") def is_alive(self): return self.kernel is not None def client(self, **kwargs): kwargs['kernel'] = self.kernel return super(InProcessKernelManager, self).client(**kwargs) #----------------------------------------------------------------------------- # ABC Registration #----------------------------------------------------------------------------- KernelManagerABC.register(InProcessKernelManager)
super(RPZKernelManager, self).finish_shutdown(*args, **kwargs) # Add the input file to the configuration config = self.rpz_options.config_file() with config.rewrite(encoding='utf-8') as (read, write): for line in read: write.write(line) if line == 'inputs_outputs:\n': write.write(' - name: jupyter_connection_file' ' # Needed for reprozip-jupyter operations\n' ' read_by_runs: [0]\n' ' path: %s\n' % self.connection_file) KernelManagerABC.register(RPZKernelManager) class RPZExecutePreprocessor(ExecutePreprocessor): def __init__(self, options): self.rpz_options = options super(RPZExecutePreprocessor, self).__init__() def preprocess(self, nb, resources): # no change { path = resources.get('metadata', {}).get('path', '') if path == '': path = None kernel_name = nb.metadata.get('kernelspec', {}).get('name', 'python') if self.kernel_name:
async def is_alive(self): """Is the kernel process still running?""" if self.has_kernel: # Go ahead and issue a request to get the kernel self.kernel = await self.get_kernel(self.kernel_id) return True else: # we don't have a kernel return False def cleanup_resources(self, restart=False): """Clean up resources when the kernel is shut down""" pass KernelManagerABC.register(HTTPKernelManager) class ChannelQueue(Queue): channel_name: str = None def __init__(self, channel_name: str, channel_socket: websocket.WebSocket, log: Logger): super().__init__() self.channel_name = channel_name self.channel_socket = channel_socket self.log = log async def get_msg(self, *args, **kwargs) -> dict: timeout = kwargs.get('timeout', 1)
async def is_alive(self): """Is the kernel process still running?""" if self.has_kernel: # Go ahead and issue a request to get the kernel self.kernel = await self.refresh_model() return True else: # we don't have a kernel return False def cleanup_resources(self, restart=False): """Clean up resources when the kernel is shut down""" pass KernelManagerABC.register(GatewayKernelManager) class ChannelQueue(Queue): channel_name: Optional[str] = None def __init__(self, channel_name: str, channel_socket: websocket.WebSocket, log: Logger): super().__init__() self.channel_name = channel_name self.channel_socket = channel_socket self.log = log async def get_msg(self, *args: Any, **kwargs: Any) -> dict: timeout = kwargs.get("timeout", 1) msg = self.get(timeout=timeout)