Fixed regression bug

mariya · mariya · commit eb71f96623fb · 2024-11-07T14:09:38.000+01:00
diff --git a/src/conformist/validation_trial.py b/src/conformist/validation_trial.py
@@ -91,7 +91,8 @@ def mean_set_sizes_by_class(self, class_names):
     def mean_fnrs_by_class(self, class_names):
         fnr_dicts = []
         for run in self.runs:
-            fnr_dicts.append(run.mean_fnrs_by_class(class_names))
+            fnr_dicts.append(run.mean_fnrs_by_class(run.prediction_sets,
+                                                    class_names))
 
         means = {}
         for class_name in class_names:
