(self, g: Graph)
| 9 | self.walk_length = walk_length |
| 10 | |
| 11 | def augment(self, g: Graph) -> Graph: |
| 12 | x, edge_index, edge_weights = g.unfold() |
| 13 | |
| 14 | edge_index, edge_weights = random_walk_subgraph(edge_index, edge_weights, batch_size=self.num_seeds, length=self.walk_length) |
| 15 | |
| 16 | return Graph(x=x, edge_index=edge_index, edge_weights=edge_weights) |
nothing calls this directly
no test coverage detected