Unverified Commit 9ef950a6 authored by Hongzhi (Steve), Chen's avatar Hongzhi (Steve), Chen Committed by GitHub
Browse files

[Graphbolt] Polish. (#6520)


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-28-63.ap-northeast-1.compute.internal>
parent 325e67ce
......@@ -19,31 +19,31 @@ tasks:
- format: numpy
in_memory: true
name: seed_nodes
path: train_node.npy
path: nodeclassification/train-node.npy
- format: numpy
in_memory: true
name: labels
path: train_label.npy
path: nodeclassification/train-label.npy
type_name: null
validation_set:
- data:
- format: numpy
in_memory: true
name: seed_nodes
path: valid_node.npy
path: nodeclassification/valid-node.npy
- format: numpy
in_memory: true
name: labels
path: valid_label.npy
path: nodeclassification/valid-label.npy
type_name: null
test_set:
- data:
- format: numpy
in_memory: true
name: seed_nodes
path: test_node.npy
path: nodeclassification/test-node.npy
- format: numpy
in_memory: true
name: labels
path: test_label.npy
path: nodeclassification/test-label.npy
type_name: null
......@@ -13,8 +13,8 @@ EXAMPLE_ROOT = os.path.join(
)
def test_gcn():
script = os.path.join(EXAMPLE_ROOT, "gcn.py")
def test_node_classification():
script = os.path.join(EXAMPLE_ROOT, "node_classification.py")
out = subprocess.run(["python", str(script)], capture_output=True)
assert out.returncode == 0
stdout = out.stdout.decode("utf-8")
......
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