Fix unnecessary trigger of lazy copy (dmlc#3585)

Co-authored-by: Mufei Li <[email protected]>
skorani · Dec 13, 2021 · 02880e9 · 02880e9
1 parent 7273a16
commit 02880e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/dgl/heterograph.py b/python/dgl/heterograph.py
@@ -4230,7 +4230,7 @@ def _get_e_repr(self, etid, edges):
         """
         # parse argument
         if is_all(edges):
-            return dict(self._edge_frames[etid])
+            return self._edge_frames[etid]
         else:
             eid = utils.parse_edges_arg_to_eid(self, edges, etid, 'edges')
             return self._edge_frames[etid].subframe(eid)