input: Write a function to implement the EpsilonGreedy class and the choose_action method.
  assert EpsilonGreedy(0.1).choose_action([0.1, 0.2, 0.3, 0.4]) in [0, 1, 2, 3]
  assert EpsilonGreedy(0.9).choose_action([0.1, 0.2, 0.3, 0.4]) in [0, 1, 2, 3]
  assert EpsilonGreedy(0.0).choose_action([0.1, 0.2, 0.3, 0.4]) == 3
output: import random
  
  class EpsilonGreedy:
      def __init__(self, epsilon):
          self.epsilon = epsilon
  
      def choose_action(self, action_values):
          if random.random() < self.epsilon:
              return random.randint(0, len(action_values) - 1)
          else:
              return action_values.index(max(action_values))