fix reducer qualname and name to support pickle (#3540)

6418c1c5 · Zeyan Li 李则言 · GitHub · 4bb94b15 · 6418c1c5
Unverified Commit 6418c1c5 authored Dec 17, 2021 by Zeyan Li 李则言 Committed by GitHub Dec 17, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

python/dgl/function/reducer.py python/dgl/function/reducer.py +2 -1

No files found.
--- a/python/dgl/function/reducer.py
+++ b/python/dgl/function/reducer.py
@@ -77,7 +77,8 @@ def _gen_reduce_builtin(reducer):

    def func(msg, out):
        return SimpleReduceFunction(reducer, msg, out)
-    func.__name__ = reducer
+    func.__name__ = str(reducer)
+    func.__qualname__ = str(reducer)
    func.__doc__ = docstring
    return func