From a35c7b8f5c1239fbc4f0c6825ea64c686af690d9 Mon Sep 17 00:00:00 2001
From: Bryan Marcus McCann <bryan.mccann.is@gmail.com>
Date: Thu, 16 Aug 2018 19:48:08 +0000
Subject: [PATCH] fixing build for predict

---
 predict.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/predict.py b/predict.py
index 541ed2d9..927e6ac0 100644
--- a/predict.py
+++ b/predict.py
@@ -110,6 +110,7 @@ def run(args, field, val_sets, model):
                     for i, pp in enumerate(p):
                         if 'sql' in task:
                             wikisql_id = int(batch.wikisql_id[i])
+                            wikisql_ids = wikisql_ids if wikisql_ids is not None else []
                             wikisql_ids.append(wikisql_id)
                         prediction_file.write(pp + '\n')
                         predictions.append(pp)