Use nbformat to set up notebook cells

jupyter · May 13, 2019 · 0c7d7ba · 0c7d7ba
1 parent 7b4a677
commit 0c7d7ba
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/nbconvert/preprocessors/tests/test_execute.py b/nbconvert/preprocessors/tests/test_execute.py
@@ -265,16 +265,7 @@ def label_parallel_notebook(nb, label):
     
     Used for parallel testing to label two notebooks which are run simultaneously.
     """
-    label_cell = nbformat.NotebookNode(
-        {
-            "cell_type": "code",
-            "execution_count": None,
-            "metadata": {},
-            "outputs": [],
-            "source": "this_notebook = '{}'".format(label),
-        }
-    )
-
+    label_cell = nbformat.v4.new_code_cell(source="this_notebook = '{}'".format(label))
     nb.cells.insert(1, label_cell)
     return nb