From c19dd2b96cc49534accf33ad811291075b831564 Mon Sep 17 00:00:00 2001
From: Nihal Nayak <nihalnayak@gmail.com>
Date: Mon, 24 Feb 2025 08:14:26 -0500
Subject: [PATCH] fix: replace {{context}} in output

---
 bonito/abstract.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bonito/abstract.py b/bonito/abstract.py
index e873c55..a002efc 100644
--- a/bonito/abstract.py
+++ b/bonito/abstract.py
@@ -102,7 +102,7 @@ def process(example):
             context = example[context_col].strip()
             return {
                 "input": pair[0].strip().replace("{{context}}", context),
-                "output": pair[1].strip(),
+                "output": pair[1].strip().replace("{{context}}", context),
             }
 
         synthetic_dataset = synthetic_dataset.map(