Merge pull request #19 from s-jse/patch-1

Fix variable name in bigbench_hard.ipynb
microsoft · Nov 23, 2024 · 1ed76f7 · 1ed76f7
2 parents 29fc5ea + 685169b
commit 1ed76f7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/examples/nlp/bigbench_hard.ipynb b/docs/examples/nlp/bigbench_hard.ipynb
@@ -794,8 +794,8 @@
    ],
    "source": [
     "task = \"sports_understanding\"\n",
-    "train = load_dataset(\"maveriq/bigbenchhard\", task)[\"train\"]\n",
-    "examples = [{\"question\": r[\"input\"], \"answer\": r[\"target\"]} for r in train]\n",
+    "train_set = load_dataset(\"maveriq/bigbenchhard\", task)[\"train\"]\n",
+    "examples = [{\"question\": r[\"input\"], \"answer\": r[\"target\"]} for r in train_set]\n",
     "\n",
     "dp = Predict()\n",
     "optimizer = OptoPrime(dp.parameters(),\n",