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