fix small typo bug (dmlc#2708)

Co-authored-by: Zihao Ye <[email protected]>
XINyexun · Feb 28, 2021 · 062f59a · 062f59a
1 parent 28b0a76
commit 062f59a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/batch.py b/python/dgl/batch.py
@@ -428,4 +428,4 @@ def unbatch_hetero(*args, **kwargs):
     """DEPREACTED: please use dgl.unbatch """
     dgl_warning('From v0.5, DGLHeteroGraph is merged into DGLGraph. You can safely'
                 ' replace dgl.unbatch_hetero with dgl.unbatch')
-    return batch(*args, **kwargs)
+    return unbatch(*args, **kwargs)