Method query_range

axlearn/common/segment_tree.py:88–92

Query operation result over range [left, right] inclusive.

(self, left: int, right: int)

            parent //= 2

    def query_range(self, left: int, right: int) -> T:
        """Query operation result over range [left, right] inclusive."""
        if left > right or left < 0 or right >= self.size:
            return self.default_value
        return self._query_range(1, 0, self.tree_size - 1, left, right)

    def _query_range(self, node: int, start: int, end: int, left: int, right: int) -> T:
        """Recursive implementation of range query."""