re-add task

logikon-ai · Apr 21, 2024 · c8bc83b · c8bc83b
1 parent 430c78a
commit c8bc83b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cot_eval/__main__.py b/src/cot_eval/__main__.py
@@ -186,7 +186,7 @@ def main():
         with tempfile.TemporaryFile() as tmpfile:
 
             df = pd.DataFrame(ds)
-            df["config_data"] = len(df) * [list(config_data.items())]
+            df["config_data"] = len(df) * [list(config_data.items()) +[("task",task)]]
             logging.info(f"Created dataframe with reasoning traces for upload:\n{df.head(3)}")
             df.to_parquet(tmpfile, index=False)