Adjusted function call for dummy run

ssciwr · Jan 4, 2025 · 36990cc · 36990cc
1 parent 37f1f6d
commit 36990cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/notebook/batching_performance.ipynb b/notebook/batching_performance.ipynb
@@ -194,10 +194,10 @@
     "# batching_sizes = [1]\n",
     "n_samples = 5\n",
     "outputs = []\n",
-    "csv_file = \"../mailcom/test/data/mails_lb_sg_copy2.csv\"\n",
+    "csv_file = \"../mailcom/test/data/mails_lb_sg.csv\"\n",
     "\n",
     "# first make a dummy run since there seem to be some inconsitencies when loading for the first time\n",
-    "_ = performance_test(-1, disp=False)\n",
+    "_ = performance_test(csv_file, -1, disp=False)\n",
     "\n",
     "# testing\n",
     "av_email_times_for_batches = []\n",