whenxuan: black init

Author: wwhenxuan

cvmd2d.ipynb

@@ -118,12 +118,14 @@
     "from pysdkit import MVMD, VMD\n",
     "from pysdkit.data import test_multivariate_signal\n",
     "\n",
-    "plt.rcParams.update({\n",
-    "    \"figure.dpi\": 120,\n",
-    "    \"font.size\": 11,\n",
-    "    \"axes.grid\": True,\n",
-    "    \"grid.alpha\": 0.25,\n",
-    "})\n",
+    "plt.rcParams.update(\n",
+    "    {\n",
+    "        \"figure.dpi\": 120,\n",
+    "        \"font.size\": 11,\n",
+    "        \"axes.grid\": True,\n",
+    "        \"grid.alpha\": 0.25,\n",
+    "    }\n",
+    ")\n",
     "\n",
     "\n",
     "def relative_error(ref, est):\n",
@@ -428,8 +430,11 @@
     "        axs[k, c].set_title(f\"Mode {k+1} | ch{c+1} | f≈{omega_hz[k]:.1f} Hz\")\n",
     "        e = mode_energies(modes[k])\n",
     "        axs[k, c].text(\n",
-    "            0.02, 0.85, f\"E=[{e[0]:.0f}, {e[1]:.0f}]\",\n",
-    "            transform=axs[k, c].transAxes, fontsize=8,\n",
+    "            0.02,\n",
+    "            0.85,\n",
+    "            f\"E=[{e[0]:.0f}, {e[1]:.0f}]\",\n",
+    "            transform=axs[k, c].transAxes,\n",
+    "            fontsize=8,\n",
     "        )\n",
     "    axs[k, 0].set_xlim(0, 0.5)\n",
     "axs[-1, 0].set_xlabel(\"Time [s]\")\n",
@@ -718,7 +723,9 @@
     "\n",
     "fig, ax = plt.subplots(figsize=(4.8, 4.2))\n",
     "im = ax.imshow(acc, vmin=0, vmax=1, cmap=\"gray_r\")\n",
-    "ax.set_title(f\"Mean |corr| of MVMD modes on {C}-ch wGn\\n(average over {n_real} realizations)\")\n",
+    "ax.set_title(\n",
+    "    f\"Mean |corr| of MVMD modes on {C}-ch wGn\\n(average over {n_real} realizations)\"\n",
+    ")\n",
     "ax.set_xlabel(\"Mode index\")\n",
     "ax.set_ylabel(\"Mode index\")\n",
     "fig.colorbar(im, ax=ax, fraction=0.046)\n",

examples/cvmd2d.ipynb

@@ -122,12 +122,14 @@
     "from pysdkit.data import test_emd\n",
     "from pysdkit.plot import plot_IMFs, plot_IMFs_amplitude_spectra\n",
     "\n",
-    "plt.rcParams.update({\n",
-    "    \"figure.dpi\": 120,\n",
-    "    \"font.size\": 11,\n",
-    "    \"axes.grid\": True,\n",
-    "    \"grid.alpha\": 0.25,\n",
-    "})\n",
+    "plt.rcParams.update(\n",
+    "    {\n",
+    "        \"figure.dpi\": 120,\n",
+    "        \"font.size\": 11,\n",
+    "        \"axes.grid\": True,\n",
+    "        \"grid.alpha\": 0.25,\n",
+    "    }\n",
+    ")\n",
     "\n",
     "\n",
     "def relative_error(ref, est):\n",
@@ -348,7 +350,7 @@
     "imfs0 = vmd0.fit_transform(s0)\n",
     "print(\"modes shape:\", imfs0.shape)\n",
     "re0 = relative_error(s0[: imfs0.shape[1]], imfs0.sum(0))\n",
-    "print(f\"relative recon. error: {re0:.4e}\")\n"
+    "print(f\"relative recon. error: {re0:.4e}\")"
    ]
   },
   {
@@ -675,7 +677,9 @@
     "fig.suptitle(\"VMD decomposition (K=3)\", y=1.01)\n",
     "plt.show()\n",
     "\n",
-    "fig = plot_IMFs(signal, imfs_emd, max_imfs=min(7, imfs_emd.shape[0]), return_figure=True)\n",
+    "fig = plot_IMFs(\n",
+    "    signal, imfs_emd, max_imfs=min(7, imfs_emd.shape[0]), return_figure=True\n",
+    ")\n",
     "fig.suptitle(f\"EMD decomposition ({imfs_emd.shape[0]} IMFs)\", y=1.01)\n",
     "plt.show()\n",
     "\n",