Move import of nbformat inside get_first_cell_content method

Deci-AI · Oct 30, 2023 · b7b55d9 · b7b55d9
1 parent 410c03f
commit b7b55d9
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/verify_notebook_version.py b/tests/verify_notebook_version.py
@@ -3,10 +3,11 @@
 from typing import Optional
 
 import super_gradients
-import nbformat
 
 
 def get_first_cell_content(notebook_path):
+    import nbformat
+
     # Load the notebook
     with open(notebook_path, "r", encoding="utf-8") as notebook_file:
         notebook_content = nbformat.read(notebook_file, as_version=4)