Don't require nbformat when skipping notebook tests

jmeyers314 · Apr 29, 2024 · b01dbef · b01dbef
1 parent 11a7091
commit b01dbef
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_notebooks.py b/tests/test_notebooks.py
@@ -4,8 +4,6 @@
 import tempfile
 from test_helpers import timer
 
-import nbformat
-
 notebook_dir = os.path.abspath(os.path.join(
     os.path.split(os.path.abspath(__file__))[0],
     '..',
@@ -18,6 +16,7 @@ def _notebook_run(path):
     """Execute a notebook via nbconvert and collect output.
        :returns (parsed nb object, execution errors)
     """
+    import nbformat
     dirname, __ = os.path.split(path)
     os.chdir(dirname)
     with tempfile.NamedTemporaryFile(suffix=".ipynb") as fout: