diff options
Diffstat (limited to 'python/function.py')
| -rw-r--r-- | python/function.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/python/function.py b/python/function.py index 39c36356..70cd6a35 100644 --- a/python/function.py +++ b/python/function.py @@ -279,10 +279,8 @@ class BasicBlockList: return self._function._instantiate_block(block) elif isinstance(i, slice): result = [] - if i.start < 0 or i.start >= len(self) or i.stop < 0 or i.stop >= len(self): - raise IndexError(f"Slice {i} out of bounds for BasicBlockList of size {len(self)}") - - for j in range(i.start, i.stop, i.step if i.step is not None else 1): + start, stop, step = i.indices(len(self)) + for j in range(start, stop, step): block = core.BNNewBasicBlockReference(self._blocks[j]) assert block is not None, "core.BNNewBasicBlockReference returned None" result.append(self._function._instantiate_block(block)) |