MCPcopy Create free account
hub / github.com/apple/axlearn / find_first

Method find_first

axlearn/common/segment_tree.py:107–111  ·  view source on GitHub ↗

Find first index >= start_index where predicate is True.

(self, predicate: Callable[[T], bool], start_index: int = 0)

Source from the content-addressed store, hash-verified

105 return self.operation(left_result, right_result)
106
107 def find_first(self, predicate: Callable[[T], bool], start_index: int = 0) -> int:
108 """Find first index >= start_index where predicate is True."""
109 if start_index >= self.size:
110 return -1
111 return self._find_first(1, 0, self.tree_size - 1, predicate, start_index)
112
113 def _find_first(
114 self, node: int, start: int, end: int, predicate: Callable[[T], bool], search_start: int

Callers 4

find_best_fitMethod · 0.80

Calls 1

_find_firstMethod · 0.95

Tested by 3