arr[end] was excluded when end is not None

royf · Apr 21, 2018 · 4e7129b · 4e7129b
1 parent 8264e64
commit 4e7129b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/ray/rllib/optimizers/segment_tree.py b/python/ray/rllib/optimizers/segment_tree.py
@@ -78,10 +78,9 @@ def reduce(self, start=0, end=None):
           elements.
         """
         if end is None:
-            end = self._capacity
+            end = self._capacity - 1
         if end < 0:
             end += self._capacity
-        end -= 1
         return self._reduce_helper(start, end, 1, 0, self._capacity - 1)
 
     def __setitem__(self, idx, val):