update

pyg-team · Dec 17, 2024 · 1f5077b · 1f5077b
1 parent 2b1b327
commit 1f5077b
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) -> str | tuple[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.: