Skip to content

GitLab

Unverified Commit 7643e537 authored by Andrei Ivanov's avatar Andrei Ivanov Committed by GitHub
Browse files

Elimination of warnings in two GraphBolt tests. (#6573)

parent 6d4958c2
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
View file @ 7643e537
......@@ -898,7 +898,7 @@ def test_OnDiskDataset_Feature_heterograph():
)
assert torch.equal(
feature_data.read("node", "paper", "labels"),
torch.tensor(node_data_label),
node_data_label.clone().detach(),
)
# Verify edge feature data.
......@@ -908,7 +908,7 @@ def test_OnDiskDataset_Feature_heterograph():
)
assert torch.equal(
feature_data.read("edge", "author:writes:paper", "labels"),
torch.tensor(edge_data_label),
edge_data_label.clone().detach(),
)
feature_data = None
......@@ -981,7 +981,7 @@ def test_OnDiskDataset_Feature_homograph():
)
assert torch.equal(
feature_data.read("node", None, "labels"),
torch.tensor(node_data_label),
node_data_label.clone().detach(),
)
# Verify edge feature data.
......@@ -991,7 +991,7 @@ def test_OnDiskDataset_Feature_homograph():
)
assert torch.equal(
feature_data.read("edge", None, "labels"),
torch.tensor(edge_data_label),
edge_data_label.clone().detach(),
)
feature_data = None
......
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