Skip to content

GitLab

Unverified Commit 4ceb0bff authored by keli-wen's avatar keli-wen Committed by GitHub
Browse files

[Graphbolt] Modify `preprocess_ondisk_dataset` function. (#6019) 

fix str2
parent fe8a8fc0
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/dgl/graphbolt/impl/ondisk_dataset.py
View file @ 4ceb0bff
...@@ -120,13 +120,13 @@ def preprocess_ondisk_dataset(input_config_path: str) -> str: ...@@ -120,13 +120,13 @@ def preprocess_ondisk_dataset(input_config_path: str) -> str:
# 5. Save the CSCSamplingGraph and modify the output_config. # 5. Save the CSCSamplingGraph and modify the output_config.
output_config["graph_topology"] = {} output_config["graph_topology"] = {}
output_config["graph_topology"]["type"] = "CSCSamplingGraph" output_config["graph_topology"]["type"] = "CSCSamplingGraph"
output_config["graph_topology"]["path"] = ( output_config["graph_topology"]["path"] = str(
processed_dir_prefix / "csc_sampling_graph.tar" processed_dir_prefix / "csc_sampling_graph.tar"
) )
save_csc_sampling_graph( save_csc_sampling_graph(
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"] del output_config["graph"]
......
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