From 452b12018345e5c2ad5acc1b1e48368163f24e1b Mon Sep 17 00:00:00 2001
From: abdullah-alnahas <abdullah.nahass@gmail.com>
Date: Wed, 15 May 2024 00:07:25 +0300
Subject: [PATCH] changing back to testing on 10 batik samples only

---
 tests/test_answer_quality.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_answer_quality.py b/tests/test_answer_quality.py
index e181805..d544fde 100644
--- a/tests/test_answer_quality.py
+++ b/tests/test_answer_quality.py
@@ -65,7 +65,7 @@ def is_correct(row):
 
 def test_ansari_agent(data):
     df, q_temp, cache = data
-    #df = df.sample(10)
+    df = df.sample(10)
     # For cost and efficiency reasons, we will only test 10 questions
     df["json_prediction"] = df.apply(
         lambda row: answer_question(row, q_temp, cache), axis=1