Small tweaks to evaluation function

test_pretrained.ipynb

@@ -329,7 +329,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 67,
+   "execution_count": 76,
    "metadata": {},
    "outputs": [
     {
@@ -344,17 +344,16 @@
      "name": "stdout",
      "output_type": "stream",
      "text": [
-      "How many times did the Minnesota Timberwolves lose at home in the 2004 season despite recording more steals and blocks than their opponent?\n",
-      "SELECT COUNT(*) FROM game g WHERE g.team_abbreviation_home = 'MIN' AND g.wl_home = 'L' AND g.stl_home > g.stl_away AND g.blk_home > g.blk_away AND g.season_id = '22004';\n",
-      "0\n",
+      "How many games did the Indiana Pacers win at home with more than 15 fast break points in 1996?\n",
+      "SELECT COUNT(*) as wins  FROM other_stats os  JOIN game g ON os.game_id = g.game_id  WHERE g.team_name_home = 'Indiana Pacers' AND g.wl_home = 'W' AND os.pts_fb_home > 15 AND g.season_id = '21996';\n",
+      "7.0\n",
       "SQLite:\n",
       "SELECT COUNT(*) \n",
       "FROM game \n",
-      "WHERE team_name_home = 'Minnesota Timberwolves' \n",
-      "AND wl_home = 'L' \n",
-      "AND season_id = '22004';\n",
+      "WHERE wl_home = 'W' \n",
+      "AND season_id = '2196' \n",
+      "AND pts_fb_home > 15;\n",
       "\n",
-      "[(17,)]\n",
       "SQL matched? False\n",
       "Result matched? False\n"
      ]
@@ -398,7 +397,7 @@
     "            result = False\n",
     "            for row in rows:\n",
     "                for r in row:\n",
-    "                    if str(r) == str(sample_result):\n",
+    "                    if str(r) in str(sample_result):\n",
     "                        return query_match, True\n",
     "\n",
     "            # Compare results and return\n",