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
...@@ -898,7 +898,7 @@ def test_OnDiskDataset_Feature_heterograph(): ...@@ -898,7 +898,7 @@ def test_OnDiskDataset_Feature_heterograph():
) )
assert torch.equal( assert torch.equal(
feature_data.read("node", "paper", "labels"), feature_data.read("node", "paper", "labels"),
torch.tensor(node_data_label), node_data_label.clone().detach(),
) )
# Verify edge feature data. # Verify edge feature data.
...@@ -908,7 +908,7 @@ def test_OnDiskDataset_Feature_heterograph(): ...@@ -908,7 +908,7 @@ def test_OnDiskDataset_Feature_heterograph():
) )
assert torch.equal( assert torch.equal(
feature_data.read("edge", "author:writes:paper", "labels"), feature_data.read("edge", "author:writes:paper", "labels"),
torch.tensor(edge_data_label), edge_data_label.clone().detach(),
) )
feature_data = None feature_data = None
...@@ -981,7 +981,7 @@ def test_OnDiskDataset_Feature_homograph(): ...@@ -981,7 +981,7 @@ def test_OnDiskDataset_Feature_homograph():
) )
assert torch.equal( assert torch.equal(
feature_data.read("node", None, "labels"), feature_data.read("node", None, "labels"),
torch.tensor(node_data_label), node_data_label.clone().detach(),
) )
# Verify edge feature data. # Verify edge feature data.
...@@ -991,7 +991,7 @@ def test_OnDiskDataset_Feature_homograph(): ...@@ -991,7 +991,7 @@ def test_OnDiskDataset_Feature_homograph():
) )
assert torch.equal( assert torch.equal(
feature_data.read("edge", None, "labels"), feature_data.read("edge", None, "labels"),
torch.tensor(edge_data_label), edge_data_label.clone().detach(),
) )
feature_data = None 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