From 593266c228d5beb3f9b97984daed1bde228495d6 Mon Sep 17 00:00:00 2001
From: Gregor Betz <3662782+ggbetz@users.noreply.github.com>
Date: Thu, 11 Apr 2024 16:01:09 +0200
Subject: [PATCH] bugfix load traces configs

---
 notebooks/CoT_Leaderboard_Results_Exploration.ipynb | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/notebooks/CoT_Leaderboard_Results_Exploration.ipynb b/notebooks/CoT_Leaderboard_Results_Exploration.ipynb
index 8560f7d..c61b25d 100644
--- a/notebooks/CoT_Leaderboard_Results_Exploration.ipynb
+++ b/notebooks/CoT_Leaderboard_Results_Exploration.ipynb
@@ -1032,8 +1032,11 @@
         "df_cottraces = pd.DataFrame(datasets.load_dataset(TRACES_DATASET)[\"test\"])\n",
         "df_cotconfigs = pd.DataFrame([\n",
         "    dict(data) for data\n",
-        "    in df_cottraces.cot_config.unique()\n",
-        "])\n"
+        "    in df_cottraces.config_data\n",
+        "    if data\n",
+        "])\n",
+        "df_cotconfigs.drop_duplicates(inplace=True, ignore_index=True)\n",
+        "df_cotconfigs"
       ]
     },
     {