OpenQA: pass the k to the evaluate_retrieval_example from the success@k

hw_openqa.ipynb

@@ -1648,7 +1648,7 @@
     "def success_at_k(examples, k=20):\n",
     "    scores = []\n",
     "    for ex in examples: \n",
-    "        scores.append(evaluate_retrieval_example(ex, k=5))\n",
+    "        scores.append(evaluate_retrieval_example(ex, k=k))\n",
     "    return sum(scores) / len(scores)\n",
     "        \n",
     "    \n",
@@ -1686,11 +1686,11 @@
     "%%time\n",
     "if torch.cuda.is_available():\n",
     "    # This will take a few hours on a CPU:\n",
-    "    print(success_at_k(squad_dev))\n",
+    "    print(success_at_k(squad_dev, k=5))\n",
     "else:\n",
     "    # This should be reasonably fast and yields the\n",
     "    # same kind of result:\n",
-    "    print(success_at_k(dev_exs))"
+    "    print(success_at_k(dev_exs, k=5))"
    ]
   },
   {