Merge branch 'master' into rebase-txt2kg

pyg-team · Dec 17, 2024 · 7768148 · 7768148
2 parents 747f0ea + ab2b458
commit 7768148
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.: