🐛 Style Transfer: remove keywords insertion at the gen dataset creation

arkhn · Nov 4, 2024 · c456dc0 · c456dc0
1 parent 898bb09
commit c456dc0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/style-transfer/style_transfer/rb_gen/steps/generate.py b/lib/style-transfer/style_transfer/rb_gen/steps/generate.py
@@ -94,7 +94,7 @@ def batch_generate(cfg, step, dataloader, llm, wb_ds_name) -> Dataset:
             )
         batch_logs = {
             "prompts": batch["query"],
-            "ground_texts": batch["text"],
+            "ground_texts": batch["ground_texts"],
         }
         batch_logs = {**batch_logs, **flattened_gs_dict}
         gen_df = pd.DataFrame.from_dict(batch_logs)