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)
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)
