Added collections import

2019-02-10 23:59:29 -05:00 · 2019-02-10 23:59:29 -05:00 · be637664e7
commit be637664e7
parent b2ab2ee132
1 changed files with 1 additions and 0 deletions
--- a/rltorch/action_selector/EpsilonGreedySelector.py
+++ b/rltorch/action_selector/EpsilonGreedySelector.py
@ -1,5 +1,6 @@
 from .ArgMaxSelector import ArgMaxSelector
 import numpy as np 
+import collections
 class EpsilonGreedySelector(ArgMaxSelector):
    def __init__(self, model, action_size, device = None, epsilon = 0.1):
        super(EpsilonGreedySelector, self).__init__(model, action_size, device = device)