fix

Author: Luigi Dello Stritto

machine_learning_hep/models.py

@@ -156,7 +156,7 @@ def apply(ml_type, names_, trainedmodels_, test_set_, mylistvariables_, labels_=
                 test_set_[f"y_test_prediction{name}"]=0
                 test_set_[f"y_test_prob{name}"]=0
             return test_set_
-        else:
+        if  ml_type == "MultiClassification":
             for name in names_:
                 for pred, lab in enumerate(labels_):
                     safe_lab = lab.replace('-', '_')

machine_learning_hep/processer.py

@@ -292,7 +292,7 @@ def cfg(self, param, default = None):
         return reduce(lambda d, key: d.get(key, default) if isinstance(d, dict) else default,
                       param.split("."), self.datap['analysis'][self.typean])
 
-    def unpack(self, file_index, max_no_keys = None): # pylint: disable=too-many-branches
+    def unpack(self, file_index, max_no_keys = None): # pylint: disable=too-many-branches, too-many-locals
         def dfread(rdir, trees, cols, idx_name=None):
             """Read DF from multiple (joinable) O2 tables"""
             try: