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

Looks good, thanks!
This commit is contained in:
Roy Fox 2018-04-22 15:12:55 -07:00 committed by Eric Liang
parent 8f59546ef2
commit baf97e450b

View file

@ -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):