bugfix

NNPDF · Apr 17, 2024 · 26b1daf · 26b1daf
1 parent 37e728f
commit 26b1daf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/n3fit/src/n3fit/hyper_optimization/rewards.py b/n3fit/src/n3fit/hyper_optimization/rewards.py
@@ -62,7 +62,7 @@ def _average_best(fold_losses: np.ndarray, percentage: float = 0.9, axis: int =
     -------
         float: The average along the specified axis.
     """
-    num_best = int(np.ceil(percentage * len(sorted_losses)))
+    num_best = int(np.ceil(percentage * len(fold_losses)))
 
     if np.isnan(fold_losses).any():
         log.warning(f"{np.isnan(fold_losses).sum()} replicas have NaNs losses")