Skip to content

GitLab

"src/git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "2a8477de5ca73d1b492e7ee5205bfcb1d7497c97"
Unverified Commit 6f9db813 authored by Mingbang Wang's avatar Mingbang Wang Committed by GitHub
Browse files

[Misc] Correct `test_OnDiskDataset_preprocess_homogeneous` (#6922)

parent de1eedc6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 4 deletions
tests/python/pytorch/graphbolt/impl/test_ondisk_dataset.py
View file @ 6f9db813
...@@ -1175,9 +1175,9 @@ def test_OnDiskDataset_preprocess_homogeneous(edge_fmt): ...@@ -1175,9 +1175,9 @@ def test_OnDiskDataset_preprocess_homogeneous(edge_fmt):
yaml_file = os.path.join(test_dir, "metadata.yaml") yaml_file = os.path.join(test_dir, "metadata.yaml")
with open(yaml_file, "w") as f: with open(yaml_file, "w") as f:
f.write(yaml_content) f.write(yaml_content)
# Test do not generate original_edge_id. # Test generating original_edge_id.
output_file = gb.ondisk_dataset.preprocess_ondisk_dataset( output_file = gb.ondisk_dataset.preprocess_ondisk_dataset(
test_dir, include_original_edge_id=False test_dir, include_original_edge_id=True
) )
with open(output_file, "rb") as f: with open(output_file, "rb") as f:
processed_dataset = yaml.load(f, Loader=yaml.Loader) processed_dataset = yaml.load(f, Loader=yaml.Loader)
...@@ -1186,8 +1186,7 @@ def test_OnDiskDataset_preprocess_homogeneous(edge_fmt): ...@@ -1186,8 +1186,7 @@ def test_OnDiskDataset_preprocess_homogeneous(edge_fmt):
) )
assert ( assert (
fused_csc_sampling_graph.edge_attributes is not None fused_csc_sampling_graph.edge_attributes is not None
and gb.ORIGINAL_EDGE_ID and gb.ORIGINAL_EDGE_ID in fused_csc_sampling_graph.edge_attributes
not in fused_csc_sampling_graph.edge_attributes
) )
fused_csc_sampling_graph = None fused_csc_sampling_graph = 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