diff --git a/rltorch/action_selector/EpsilonGreedySelector.py b/rltorch/action_selector/EpsilonGreedySelector.py index cbabf64..e537f2f 100644 --- a/rltorch/action_selector/EpsilonGreedySelector.py +++ b/rltorch/action_selector/EpsilonGreedySelector.py @@ -9,5 +9,5 @@ class EpsilonGreedySelector(ArgMaxSelector): # best_act is already implemented in ArgMaxSelector def act(self, state): eps = next(self.epsilon) if isinstance(self.epsilon, collections.Iterable) else self.epsilon - action = self.random_act() if np.random.rand() < epsilon else self.best_act(state) + action = self.random_act() if np.random.rand() < eps else self.best_act(state) return action \ No newline at end of file