pub_soft/dgl
3
0
Fork 0
mirror of https://github.com/dmlc/dgl.git synced 2026-06-03 19:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Graphbolt] Modify preprocess_ondisk_dataset function. (#6019)

Browse Source 
fix str2
This commit is contained in:
keli-wen
2023-07-19 16:12:10 +08:00
committed by GitHub
parent fe8a8fc0f6
commit 4ceb0bffd7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/dgl/graphbolt/impl/ondisk_dataset.py
View File

@@ -120,13 +120,13 @@ def preprocess_ondisk_dataset(input_config_path: str) -> str:
# 5. Save the CSCSamplingGraph and modify the output_config.
output_config["graph_topology"] = {}
output_config["graph_topology"]["type"] = "CSCSamplingGraph"
output_config["graph_topology"]["path"] = (
output_config["graph_topology"]["path"] = str(
processed_dir_prefix / "csc_sampling_graph.tar"
)
save_csc_sampling_graph(
csc_sampling_graph,
dataset_path / output_config["graph_topology"]["path"],
str(dataset_path / output_config["graph_topology"]["path"]),
)
del output_config["graph"]