def collator(self, items): result = super().collate(items) # Copy the feature from parent graph _pop_subgraph_storage(result[1], self.g) _pop_subgraph_storage(result[2], self.g) _pop_storages(result[-1], self.g_sampling) return result
def collator(self, items): """ The interface of collator, input items is edge id of the attached graph """ result = super().collate(items) # Copy the feature from parent graph _pop_subgraph_storage(result[1], self.g) _pop_subgraph_storage(result[2], self.g) _pop_storages(result[-1], self.g_sampling) return result