<reponame>AJB0211/BanditSim
from BanditSim.util import dictmax


def greedy_action(self):
    return dictmax(self.Qs)

from BanditSim.util import dictmax


def greedy_action(self):
    return dictmax(self.Qs)