changing back to testing on 10 batik samples only

ansari-project · May 14, 2024 · 452b120 · 452b120
1 parent 35ba60e
commit 452b120
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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