diff --git a/chainerrl/misc/collections.py b/chainerrl/misc/collections.py index b70a71d84..2b125ed8a 100644 --- a/chainerrl/misc/collections.py +++ b/chainerrl/misc/collections.py @@ -107,6 +107,4 @@ def popleft(self): return self._queue_front.pop() def sample(self, k): - nf = len(self._queue_front) - return [self._queue_front[i] if i < nf else self._queue_back[i - nf] - for i in sample_n_k(len(self), k)] + return [self[i] for i in sample_n_k(len(self), k)]