From 2ae590edf41a5bc168360e9186040e839395467d Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Thu, 16 Jan 2025 09:57:00 +0100
Subject: [PATCH] Rename the benchmark dataset split from train to test (#216)

---
 examples/benchmark.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/benchmark.ipynb b/examples/benchmark.ipynb
index fb65d3b..1ff00d4 100644
--- a/examples/benchmark.ipynb
+++ b/examples/benchmark.ipynb
@@ -181,7 +181,7 @@
     "import datasets\n",
     "import pandas as pd\n",
     "\n",
-    "eval_ds = datasets.load_dataset(\"m-ric/smol_agents_benchmark\")[\"train\"]\n",
+    "eval_ds = datasets.load_dataset(\"m-ric/smol_agents_benchmark\")[\"test\"]\n",
     "pd.DataFrame(eval_ds)"
    ]
   },