Ensured all classifier tasks are given a probability column name

dessn · Jul 1, 2022 · f343dc5 · f343dc5
1 parent f8cfb2d
commit f343dc5
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/pippin/aggregator.py b/pippin/aggregator.py
@@ -90,9 +90,8 @@ def __init__(self, name, output_dir, config, dependencies, options, recal_aggtas
             Task.fail_config(f"Attempting to find python file {self.python_file} but it's not there!")
 
         merge_classifiers = self.config.get("MERGE_CLASSIFIERS")
-        if merge_classifiers is None:
-            self.classifier_merge = {c.output['name']: c.get_prob_column_name() for c in self.classifiers}
-        else:
+        self.classifier_merge = {c.output['name']: c.get_prob_column_name() for c in self.classifiers}
+        if merge_classifiers is not None:
             self.classifier_merge = dict()
             for c in self.classifiers:
                 prob_col = []