Skip to content

GitLab

Unverified Commit 35653ddd authored by VoVAllen's avatar VoVAllen Committed by GitHub
Browse files

[Fix] Fix karate due to networkx version change

parent 5fc7eb6d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
python/dgl/data/karate.py
View file @ 35653ddd
...@@ -18,7 +18,7 @@ class KarateClub(object): ...@@ -18,7 +18,7 @@ class KarateClub(object):
def __init__(self): def __init__(self):
kG = nx.karate_club_graph() kG = nx.karate_club_graph()
self.label = np.array( self.label = np.array(
[kG.node[i]['club'] != 'Mr. Hi' for i in kG.nodes]).astype(np.int64) [kG.nodes[i]['club'] != 'Mr. Hi' for i in kG.nodes]).astype(np.int64)
g = DGLGraph(kG) g = DGLGraph(kG)
g.ndata['label'] = self.label g.ndata['label'] = self.label
self.data = [g] self.data = [g]
......
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