Skip to content

Commit

Permalink
Only clean the kernel client
Browse files Browse the repository at this point in the history
  • Loading branch information
martinRenou committed Jun 26, 2023
1 parent 54d268d commit 096e25f
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions voila/notebook_renderer.py
Original file line number Diff line number Diff line change
Expand Up @@ -244,6 +244,8 @@ async def _jinja_notebook_execute(self, nb, kernel_id):
# (it seems to be local to our block)
nb.cells = result.cells

await self._cleanup_resources()

async def _jinja_cell_generator(self, nb, kernel_id):
"""Generator that will execute a single notebook cell at a time"""
nb, _ = ClearOutputPreprocessor().preprocess(
Expand Down Expand Up @@ -293,6 +295,11 @@ async def _jinja_cell_generator(self, nb, kernel_id):
finally:
yield output_cell

await self._cleanup_resources()

async def _cleanup_resources(self):
await ensure_async(self.executor.kc.stop_channels())

async def load_notebook(self, path):
model = await ensure_async(self.contents_manager.get(path=path))
if "content" not in model:
Expand Down

0 comments on commit 096e25f

Please sign in to comment.