Filtering out answers not present in data does not happen (run-llama#…

…7926)
Arize-ai · Oct 2, 2023 · 4f1ac38 · 4f1ac38
1 parent 17f905e
commit 4f1ac38
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/examples/finetuning/knowledge/finetune_knowledge.ipynb b/docs/examples/finetuning/knowledge/finetune_knowledge.ipynb
@@ -290,7 +290,7 @@
     "        )\n",
     "\n",
     "        print(f\"[{idx}] QA Pair: {qa_pair} \\n Eval: {eval}\")\n",
-    "        if \"NO\" in eval:\n",
+    "        if \"NO\" in str(eval):\n",
     "            continue\n",
     "        else:\n",
     "            # new_lines.append(line)\n",