Merge branch 'master' of personal:michailmelonas/pytorch_geometric in…

…to add-token-gt
pyg-team · Dec 18, 2024 · c1f55a0 · c1f55a0
2 parents 58fa8ed + ab2b458
commit c1f55a0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/torch_geometric/typing.py b/torch_geometric/typing.py
@@ -345,6 +345,9 @@ def to_tuple(self) -> EdgeType:
                              f"tuple since it holds invalid characters")
         return self.edge_type
 
+    def __reduce__(self) -> tuple[Any, Any]:
+        return (self.__class__, (self.edge_type, ))
+
 
 # There exist some short-cuts to query edge-types (given that the full triplet
 # can be uniquely reconstructed, e.g.: