|
11 | 11 | import os
|
12 | 12 | import sys
|
13 | 13 | import traceback
|
| 14 | +from functools import partial |
14 | 15 | from copy import deepcopy
|
15 | 16 | from typing import Generator, List, Tuple, Union
|
16 | 17 |
|
@@ -154,23 +155,21 @@ def generate_content_generator(
|
154 | 155 | kernel_id: Union[str, None] = None,
|
155 | 156 | kernel_future=None,
|
156 | 157 | ) -> Generator:
|
157 |
| - async def inner_kernel_start(nb): |
158 |
| - return await self._jinja_kernel_start(nb, kernel_id, kernel_future) |
159 |
| - |
160 |
| - def inner_cell_generator(nb, kernel_id): |
161 |
| - return self._jinja_cell_generator(nb, kernel_id) |
| 158 | + inner_kernel_start = partial( |
| 159 | + self._jinja_kernel_start, kernel_id=kernel_id, kernel_future=kernel_future |
| 160 | + ) |
162 | 161 |
|
163 |
| - # These functions allow the start of a kernel and execution of the |
164 |
| - # notebook after (parts of) the template has been rendered and send |
| 162 | + # kernel_start and cell_generator allow starting the kernel and executing the |
| 163 | + # notebook after (parts of) the template has been rendered and sent |
165 | 164 | # to the client to allow progressive rendering.
|
166 | 165 | # Template should first call kernel_start, and then decide to use
|
167 |
| - # notebook_executer cell_generator to implement progressive cell rendering |
| 166 | + # notebook_executer cell_generator to implement progressive cell rendering. |
168 | 167 |
|
169 | 168 | extra_context = {
|
170 | 169 | "frontend": "voila",
|
171 | 170 | "main_js": "voila.js",
|
172 | 171 | "kernel_start": inner_kernel_start,
|
173 |
| - "cell_generator": inner_cell_generator, |
| 172 | + "cell_generator": self._jinja_cell_generator, |
174 | 173 | "notebook_execute": self._jinja_notebook_execute,
|
175 | 174 | }
|
176 | 175 | # render notebook in snippets, then return an iterator so we can flush
|
|
0 commit comments