REF: simplify and remove code surrounding padding

ChristopherMayes · Dec 16, 2024 · 5e62cb1 · 5e62cb1
1 parent eee53d7
commit 5e62cb1
Show file tree

Hide file tree

Showing 3 changed files with 295 additions and 239 deletions.
diff --git a/docs/examples/wavefront.ipynb b/docs/examples/wavefront.ipynb
@@ -358,12 +358,12 @@
     "\n",
     "domain_x = X.rspace_domain[1]\n",
     "\n",
-    "X.drift(0.0, inplace=True)\n",
+    "X = X.drift(0.0)\n",
     "\n",
     "for zi in range(0, num_drift_steps):\n",
     "    if zi > 0:\n",
     "        print(\"Propagating to: \", zi * dz)\n",
-    "        X.drift(dz, inplace=True)\n",
+    "        X = X.drift(dz)\n",
     "    wf = np.abs(get_transverse_slice(X.rmesh)) ** 2\n",
     "\n",
     "    popt_gaussian, ydata_fit, FWHM, roots = gaussian_fit(\n",