diff --git a/rltorch/action_selector/EpsilonGreedySelector.py b/rltorch/action_selector/EpsilonGreedySelector.py index ec92498..d196c1b 100644 --- a/rltorch/action_selector/EpsilonGreedySelector.py +++ b/rltorch/action_selector/EpsilonGreedySelector.py @@ -9,7 +9,7 @@ class EpsilonGreedySelector(ArgMaxSelector): # random_act is already implemented in ArgMaxSelector # best_act is already implemented in ArgMaxSelector def act(self, state): - action = self.random_act() if np.random.rand() < self.epsilon else self.best_act() + action = self.random_act() if np.random.rand() < self.epsilon else self.best_act(state) if self.epsilon > self.epsilon_min: self.epsilon = self.epsilon * self.epsilon_decay return action \ No newline at end of file