fix bug in 'ModelConf.py' (#81)

microsoft · Oct 18, 2019 · 9d48764 · 9d48764
1 parent b05ef8b
commit 9d48764
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ModelConf.py b/ModelConf.py
@@ -285,6 +285,7 @@ def configurate_data_format(self):
  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: