From be637664e7867259d900d14d4b13158b92172f7f Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Sun, 10 Feb 2019 23:59:29 -0500 Subject: [PATCH] Added collections import --- rltorch/action_selector/EpsilonGreedySelector.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rltorch/action_selector/EpsilonGreedySelector.py b/rltorch/action_selector/EpsilonGreedySelector.py index 7dbc4ae..cbabf64 100644 --- 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)