From 430c78ae1a1329224e27fd908edece043770bad2 Mon Sep 17 00:00:00 2001 From: Gregor Betz <3662782+ggbetz@users.noreply.github.com> Date: Sun, 21 Apr 2024 09:39:51 +0200 Subject: [PATCH] bugfic config_data --- src/cot_eval/__main__.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/cot_eval/__main__.py b/src/cot_eval/__main__.py index d79d10c..1767a65 100644 --- 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())+[("task",task)]] + df["config_data"] = len(df) * [list(config_data.items())] logging.info(f"Created dataframe with reasoning traces for upload:\n{df.head(3)}") df.to_parquet(tmpfile, index=False)