update

.vscode/settings.json

@@ -5,11 +5,9 @@
     },
     "editor.defaultFormatter": "ms-python.black-formatter"
   },
-  "[jupyter]": {
-    "editor.codeActionsOnSave": {
-      "source.organizeImports": "always"
-    },
-    "editor.defaultFormatter": "ms-python.black-formatter"
+  "notebook.defaultFormatter": "ms-python.black-formatter",
+  "notebook.codeActionsOnSave": {
+    "source.organizeImports": "explicit"
   },
   "[cpp]": {
     "editor.defaultFormatter": "llvm-vs-code-extensions.vscode-clangd"

notebooks/example.ipynb

@@ -7,9 +7,11 @@
    "outputs": [],
    "source": [
     "import myproj.simulation as ms\n",
+    "import numpy as np\n",
     "\n",
     "print(ms.add(1, 2))\n",
-    "print(ms.complex(1, 2))"
+    "print(ms.complex(1, 2))\n",
+    "print(np.array([1, 2, 3]))"
    ]
   }
  ],