Skip to content

GitLab

Unverified Commit 6418c1c5 authored by Zeyan Li 李则言's avatar Zeyan Li 李则言 Committed by GitHub
Browse files

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

parent 4bb94b15
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
python/dgl/function/reducer.py
View file @ 6418c1c5
...@@ -77,7 +77,8 @@ def _gen_reduce_builtin(reducer): ...@@ -77,7 +77,8 @@ def _gen_reduce_builtin(reducer):
def func(msg, out): def func(msg, out):
return SimpleReduceFunction(reducer, msg, out) return SimpleReduceFunction(reducer, msg, out)
func.__name__ = reducer func.__name__ = str(reducer)
func.__qualname__ = str(reducer)
func.__doc__ = docstring func.__doc__ = docstring
return func return func
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment