"examples/cpp/git@developer.sourcefind.cn:OpenDAS/vision.git" did not exist on "508bc1dc8d7cff5c1383068d6601ff669f69111d"
Unverified Commit c4791fd4 authored by Rhett Ying's avatar Rhett Ying Committed by GitHub
Browse files

[bugfix] fix default ntypes/etypes consistency between dgl.DGLGraph and dgl.graph (#3198)

parent 2583ec59
...@@ -39,8 +39,8 @@ class DGLHeteroGraph(object): ...@@ -39,8 +39,8 @@ class DGLHeteroGraph(object):
# pylint: disable=unused-argument, dangerous-default-value # pylint: disable=unused-argument, dangerous-default-value
def __init__(self, def __init__(self,
gidx=[], gidx=[],
ntypes=['_U'], ntypes=['_N'],
etypes=['_V'], etypes=['_E'],
node_frames=None, node_frames=None,
edge_frames=None, edge_frames=None,
**deprecate_kwargs): **deprecate_kwargs):
......
...@@ -369,6 +369,13 @@ def test_is_sorted(): ...@@ -369,6 +369,13 @@ def test_is_sorted():
assert src_sorted == True assert src_sorted == True
assert dst_sorted == False assert dst_sorted == False
def test_default_types():
dg = dgl.DGLGraph()
g = dgl.graph(([], []))
assert dg.ntypes == g.ntypes
assert dg.etypes == g.etypes
if __name__ == '__main__': if __name__ == '__main__':
test_query() test_query()
test_mutation() test_mutation()
...@@ -377,3 +384,4 @@ if __name__ == '__main__': ...@@ -377,3 +384,4 @@ if __name__ == '__main__':
test_find_edges() test_find_edges()
test_hypersparse_query() test_hypersparse_query()
test_is_sorted() test_is_sorted()
test_default_types()
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