From f1a52cf77b6de5cbb14255df8de17d3bebe053f7 Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Sun, 3 Feb 2019 21:55:08 -0500 Subject: [PATCH] Added missing dependency --- rltorch/action_selector/EpsilonGreedySelector.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/rltorch/action_selector/EpsilonGreedySelector.py b/rltorch/action_selector/EpsilonGreedySelector.py index 7bb04c1..ec92498 100644 --- a/rltorch/action_selector/EpsilonGreedySelector.py +++ b/rltorch/action_selector/EpsilonGreedySelector.py @@ -1,4 +1,5 @@ -from .ArgMaxSelector import ArgMaxSelector +from .ArgMaxSelector import ArgMaxSelector +import numpy as np class EpsilonGreedySelector(ArgMaxSelector): def __init__(self, model, action_size, device = None, epsilon = 0.1, epsilon_decay = 1, epsilon_min = 0.1): super(EpsilonGreedySelector, self).__init__(model, action_size, device = device)