[doc] fix notebook benchmark

Author: johannes-moegerle

docs/examples/benchmark/benchmark_njit.ipynb

@@ -29,7 +29,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": null,
+   "execution_count": 2,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -45,13 +45,18 @@
     "    state = rydstate.RydbergStateSQDTAlkali(species, n, l=l, j=l + 0.5)\n",
     "    potential = state.potential\n",
     "    nu = state.nu\n",
+    "    RadialKet.clear_cached_instances()\n",
     "    radial = RadialKet(nu, potential, dz=1e-3, use_njit=use_njit)\n",
     "    radial.integrate_wavefunction()\n",
     "\n",
     "    results = []\n",
     "    for species, n, l, use_njit in test_cases:\n",
     "        # Setup the test function\n",
-    "        stmt = \"radial = RadialKet(nu, potential, dz=1e-3, use_njit=use_njit)\\nradial.integrate_wavefunction()\"\n",
+    "        stmt = (\n",
+    "            \"RadialKet.clear_cached_instances()\\n\"\n",
+    "            \"radial = RadialKet(nu, potential, dz=1e-3, use_njit=use_njit)\\n\"\n",
+    "            \"radial.integrate_wavefunction()\"\n",
+    "        )\n",
     "\n",
     "        # Time the integration multiple times and take average/std\n",
     "        globals_dict = {\n",
@@ -92,8 +97,8 @@
       "----------------------------------------------------------------------\n",
       " species   n   l   use_njit  time (ms)   std (ms)\n",
       "----------------------------------------------------------------------\n",
-      "       H 100  80       True      14.74       0.96\n",
-      "       H 100  80      False     140.28       4.93\n"
+      "       H 100  80       True      13.75       0.20\n",
+      "       H 100  80      False     136.34      10.99\n"
      ]
     }
    ],