Fixed bug in email-averaged time calculation

notebook/batching_performance.ipynb

@@ -156,26 +156,24 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# batching_sizes = [-1, 1, 2, 3, 4, 6, 8, 10]\n",
-    "batching_sizes = [1, 10]\n",
-    "n_samples = 3\n",
+    "batching_sizes = [-1, 1, 2, 3, 4, 6, 8, 10]\n",
+    "# batching_sizes = [1, 10]\n",
+    "n_samples = 5\n",
     "outputs = []\n",
-    "csv_file = \"../mailcom/test/data/mails_lb_sg_copy.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(csv_file, -1)\n",
     "\n",
     "# testing\n",
     "for bs in batching_sizes:\n",
-    "    average_email_time = 0\n",
     "    for sid in range(n_samples):\n",
     "        t, out = performance_test(csv_file, bs)\n",
-    "        average_email_time += t\n",
     "        test_result_dict = {\n",
     "            \"batch_size\": bs,\n",
     "            \"sample\": sid,\n",
     "            \"email_outputs\": out,\n",
-    "            \"average_email_time\": average_email_time\n",
+    "            \"average_email_time\": t\n",
     "        }\n",
     "        outputs.append(test_result_dict)\n"
    ]