Fixing local predictions for ensembles and fusions input_fields

bigmlcom · Jun 6, 2019 · 2e33082 · 2e33082
1 parent d891e72
commit 2e33082
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/bigml/ensemble.py b/bigml/ensemble.py
@@ -166,6 +166,7 @@ def __init__(self, ensemble,
                 self.objective_id = ensemble['object'].get("objective_field")
                 query_string = EXCLUDE_FIELDS
                 no_check_fields = True
+        self.input_fields = ensemble['object'].get('input_fields')
 
         number_of_models = len(models)
         if max_models is None:

diff --git a/bigml/fusion.py b/bigml/fusion.py
@@ -164,6 +164,7 @@ def __init__(self, fusion, api=None, max_models=None):
             self.fields = fusion.get( \
                 'fusion', {}).get("fields")
             self.objective_id = fusion.get("objective_field")
+        self.input_fields = fusion.get("input_fields")
 
         number_of_models = len(self.model_ids)