[rllib] arr[end] was excluded when end is not None (ray-project#1931)

Looks good, thanks!
LoganDing · Apr 22, 2018 · baf97e4 · baf97e4
1 parent 8f59546
commit baf97e4
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):