diff --git a/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py b/package/samplers/mab_epsilon_greedy/mab_epsilon_greedy.py index d10c5668..796d6dc6 100644 --- 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], )