refactor: Delete unnecessary max operation.

optuna · Oct 11, 2024 · e4ad3bb · e4ad3bb
1 parent f806fbb
commit e4ad3bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py b/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py
@@ -62,10 +62,10 @@ def sample_independent(
             if study.direction == StudyDirection.MINIMIZE:
                 return min(
                     param_distribution.choices,
-                    key=lambda x: rewards_by_choice[x] / max(cnt_by_choice[x], 1),
+                    key=lambda x: rewards_by_choice[x] / cnt_by_choice[x],
                 )
             else:
                 return max(
                     param_distribution.choices,
-                    key=lambda x: rewards_by_choice[x] / max(cnt_by_choice[x], 1),
+                    key=lambda x: rewards_by_choice[x] / cnt_by_choice[x],
                 )