Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "a080f0d3a21104c19effb4d9d95dcb710d99c19b"
Unverified Commit bc72349c authored by peizhou001's avatar peizhou001 Committed by GitHub
Browse files

[Graohbolt] Fix CI bug (#6224) 


Co-authored-by: default avatarUbuntu <ubuntu@ip-172-31-16-19.ap-northeast-1.compute.internal>
parent d82cfdc0
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 1 deletion
tests/python/pytorch/graphbolt/test_feature_fetcher.py
View file @ bc72349c
...@@ -21,7 +21,6 @@ def test_FeatureFetcher_homo(): ...@@ -21,7 +21,6 @@ def test_FeatureFetcher_homo():
fanouts = [torch.LongTensor([2]) for _ in range(num_layer)] fanouts = [torch.LongTensor([2]) for _ in range(num_layer)]
data_block_converter = Mapper(minibatch_dp, gb_test_utils.to_node_block) data_block_converter = Mapper(minibatch_dp, gb_test_utils.to_node_block)
sampler_dp = gb.NeighborSampler(data_block_converter, graph, fanouts) sampler_dp = gb.NeighborSampler(data_block_converter, graph, fanouts)
keys.append(("node", None, "c"))
fetcher_dp = gb.FeatureFetcher(sampler_dp, feature_store, keys) fetcher_dp = gb.FeatureFetcher(sampler_dp, feature_store, keys)
assert len(list(fetcher_dp)) == 5 assert len(list(fetcher_dp)) == 5
......
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