From 9d48764dbe3ca12d8d949bf2b2dcd869e08aefc7 Mon Sep 17 00:00:00 2001
From: woailaosang <2284728029@qq.com>
Date: Fri, 18 Oct 2019 21:07:53 +0800
Subject: [PATCH] fix bug in 'ModelConf.py' (#81)

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

diff --git a/ModelConf.py b/ModelConf.py
index bb77273..9b170d0 100644
--- a/ModelConf.py
+++ b/ModelConf.py
@@ -285,6 +285,7 @@ class ModelConf(object):
                 if self.file_columns is None:
                     self.raise_configuration_error('file_columns')
             if self.phase == 'predict':
+                self.file_columns, self.predict_file_columns = self.file_header, self.predict_file_header
                 if self.file_columns is None and self.predict_file_columns is None:
                     self.raise_configuration_error('predict_file_columns')
                 if self.file_columns and self.predict_file_columns is None: