MAINT: re-run notebook after removal of .drift/.focus

docs/examples/wavefront.ipynb

@@ -13,7 +13,7 @@
     "from scipy.optimize import curve_fit\n",
     "from scipy.interpolate import UnivariateSpline\n",
     "\n",
-    "from pmd_beamphysics.wavefront import Wavefront"
+    "from pmd_beamphysics.wavefront import Wavefront, drift"
    ]
   },
   {
@@ -249,7 +249,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "drifted_w = W.drift(3.0)\n",
+    "drifted_w = drift(W, 3.0)\n",
     "fig, (ax1, ax2) = plt.subplots(ncols=2, figsize=(10, 4))\n",
     "\n",
     "rspace_abs_orig = np.abs(get_transverse_slice(W.rmesh)) ** 2\n",
@@ -329,12 +329,12 @@
     "\n",
     "domain_x = X.rspace_domain[1]\n",
     "\n",
-    "X = X.drift(0.0)\n",
+    "X = drift(X, 0.0)\n",
     "\n",
     "for zi in range(0, num_drift_steps):\n",
     "    if zi > 0:\n",
     "        print(\"Propagating to: \", zi * dz)\n",
-    "        X = X.drift(dz)\n",
+    "        X = drift(X, dz)\n",
     "    wf = np.abs(get_transverse_slice(X.rmesh)) ** 2\n",
     "\n",
     "    popt_gaussian, ydata_fit, FWHM, roots = gaussian_fit(\n",