[BasicTasks_pytorch.ipynb] Run time error: Expected type graph.Graph but get graph.HeteroGraph

[jianjindong]

Hi there, when I ran the BasicTasks_pytorch, everything is working well just except the last code block.
The error I got is as below:
DGLError: [23:05:38] /opt/dgl/include/dgl/packed_func_ext.h:117: Check failed: ObjectTypeChecker::Check(sptr.get()): Expected type graph.Graph but get graph.HeteroGraph
Do you know where it goes wrong?
Thank you.