bugfix load traces configs

logikon-ai · Apr 11, 2024 · 593266c · 593266c
1 parent 1c32545
commit 593266c
Showing 1 changed file with 5 additions and 2 deletions.
diff --git 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"
       ]
     },
     {