Fixed regression bug

Molmed · Nov 7, 2024 · eb71f96 · eb71f96
1 parent 3d5b380
commit eb71f96
Showing 1 changed file with 2 additions and 1 deletion.
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: