| 416 | # This class exists to avoid recursion into the feature dictionary returned by the |
| 417 | # prefetcher when calling recursive_apply(). |
| 418 | class _PrefetchedGraphFeatures(object): |
| 419 | __slots__ = ["node_feats", "edge_feats"] |
| 420 | |
| 421 | def __init__(self, node_feats, edge_feats): |
| 422 | self.node_feats = node_feats |
| 423 | self.edge_feats = edge_feats |
| 424 | |
| 425 | |
| 426 | def _prefetch_for_subgraph(subg, dataloader): |
no outgoing calls
no test coverage detected