From 27398f9364042bad130ca13ee11606640b821f0d Mon Sep 17 00:00:00 2001
From: Eric Hartford <ehartford@users.noreply.huggingface.co>
Date: Sat, 17 Jun 2023 12:50:34 +0000
Subject: [PATCH] Update FLAN-5m.ipynb

---
 FLAN-5m.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/FLAN-5m.ipynb b/FLAN-5m.ipynb
index 968c138..9a49c15 100644
--- a/FLAN-5m.ipynb
+++ b/FLAN-5m.ipynb
@@ -82,7 +82,7 @@
     "    f = open(\"data/cot.jsonl\", \"w\", encoding='utf8')\n",
     "    stream = tqdm.tqdm(cot, total=cot_total)  \n",
     "    for i, data in enumerate(stream):\n",
-    "        if data['template_type'] != 'zs_opt':\n",
+    "        if not 'zs' in data['template_type']:\n",
     "            continue\n",
     "        question = data['inputs']\n",
     "        system_prompt = sm_cot()\n",