(self, g: Graph)
| 8 | self.pf = pf |
| 9 | |
| 10 | def augment(self, g: Graph) -> Graph: |
| 11 | x, edge_index, edge_weights = g.unfold() |
| 12 | x = dropout_feature(x, self.pf) |
| 13 | return Graph(x=x, edge_index=edge_index, edge_weights=edge_weights) |
nothing calls this directly
no test coverage detected