bugfix in .training_features- thanks test suite!

ClimbsRocks · Feb 22, 2018 · aacf6e0 · aacf6e0
1 parent e6650b5
commit aacf6e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/auto_ml/predictor.py b/auto_ml/predictor.py
@@ -307,7 +307,7 @@ def _prepare_for_training(self, X):
         clean_descriptions = {}
         col_names = set(X_df.columns)
         for k, v in self.column_descriptions.items():
-            if k in col_names or '_day_part' in k:
+            if k in col_names or '_day_part' in k or v == 'output':
                 clean_descriptions[k] = v
         self.column_descriptions = clean_descriptions