Find the set of edge indices to remove.
(self, parent_eids)
| 53 | self._filter = recursive_apply(exclude_eids, utils.Filter) |
| 54 | |
| 55 | def _find_indices(self, parent_eids): |
| 56 | """Find the set of edge indices to remove.""" |
| 57 | if self._exclude_eids is not None: |
| 58 | parent_eids_np = recursive_apply(parent_eids, F.zerocopy_to_numpy) |
| 59 | return _locate_eids_to_exclude(parent_eids_np, self._exclude_eids) |
| 60 | else: |
| 61 | assert self._filter is not None |
| 62 | func = lambda x, y: x.find_included_indices(y) |
| 63 | return recursive_apply_pair(self._filter, parent_eids, func) |
| 64 | |
| 65 | def __call__(self, frontier, weights=None): |
| 66 | parent_eids = frontier.edata[EID] |
no test coverage detected