format

Alnusjaponica · Mar 28, 2024 · 3981f69 · 3981f69
1 parent f908912
commit 3981f69
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/optuna/pruners/_wilcoxon.py b/optuna/pruners/_wilcoxon.py
@@ -198,10 +198,14 @@ def prune(self, study: "optuna.study.Study", trial: FrozenTrial) -> bool:
 
         if study.direction == StudyDirection.MAXIMIZE:
             alt = "less"
-            average_is_best = sum(best_step_values) / len(best_step_values) <= sum(step_values) / len(step_values)
+            average_is_best = sum(best_step_values) / len(best_step_values) <= sum(
+                step_values
+            ) / len(step_values)
         else:
             alt = "greater"
-            average_is_best = sum(best_step_values) / len(best_step_values) >= sum(step_values) / len(step_values)
+            average_is_best = sum(best_step_values) / len(best_step_values) >= sum(
+                step_values
+            ) / len(step_values)
 
         # We use zsplit to avoid the problem when all values are zero.
         p = ss.wilcoxon(diff_values, alternative=alt, zero_method="zsplit").pvalue