From baf97e450b684fb1c54db8bd1c40a3c575732332 Mon Sep 17 00:00:00 2001 From: Roy Fox Date: Sun, 22 Apr 2018 15:12:55 -0700 Subject: [PATCH] [rllib] arr[end] was excluded when end is not None (#1931) Looks good, thanks! --- python/ray/rllib/optimizers/segment_tree.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/ray/rllib/optimizers/segment_tree.py b/python/ray/rllib/optimizers/segment_tree.py index 3409aa667..b412a89bd 100644 --- a/python/ray/rllib/optimizers/segment_tree.py +++ b/python/ray/rllib/optimizers/segment_tree.py @@ -78,10 +78,9 @@ class SegmentTree(object): 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):