Avoid transforming original data

This PR clones the original data object in `ToUndirected` so that the method does not change the original data
pyg-team · Jun 7, 2023 · 6930b7e · 6930b7e
1 parent 3b545ee
commit 6930b7e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_geometric/transforms/to_undirected.py b/torch_geometric/transforms/to_undirected.py
@@ -38,6 +38,7 @@ def forward(
         self,
         data: Union[Data, HeteroData],
     ) -> Union[Data, HeteroData]:
+        data = data.clone()
         for store in data.edge_stores:
             if 'edge_index' not in store:
                 continue