From e4d1e218ff04b137af1f691c30b323965f9a7007 Mon Sep 17 00:00:00 2001 From: jalencato Date: Wed, 18 Dec 2024 19:00:39 -0800 Subject: [PATCH] Update graphbolt-graph-construction.sh --- .../graphbolt-graph-construction.sh | 15 ++------------- 1 file changed, 2 insertions(+), 13 deletions(-) diff --git a/tests/end2end-tests/graphbolt-gs-integration/graphbolt-graph-construction.sh b/tests/end2end-tests/graphbolt-gs-integration/graphbolt-graph-construction.sh index 60edde579..c73f197b8 100644 --- a/tests/end2end-tests/graphbolt-gs-integration/graphbolt-graph-construction.sh +++ b/tests/end2end-tests/graphbolt-gs-integration/graphbolt-graph-construction.sh @@ -126,7 +126,8 @@ python3 -m graphstorm.gpartition.dist_partition_graph \ --num-parts 2 \ --output-path "$DIST_GRAPHBOLT_PATH" \ --ssh-port 2222 \ - --use-graphbolt "true" + --use-graphbolt "true" \ + --process-group-timeout 3600 # Ensure GraphBolt files were created by GSPartition for i in $(seq 0 1); do @@ -137,16 +138,4 @@ for i in $(seq 0 1); do fi done -echo "********* Test GSPartition with process-group-timeout ********" - -DIST_GRAPHBOLT_PATH="${OUTPUT_PATH}/graphbolt-gspartition-nc" -python3 -m graphstorm.gpartition.dist_partition_graph \ - --input-path "${INPUT_PATH}" \ - --ip-config ip_list.txt \ - --metadata-filename chunked_graph_meta.json \ - --num-parts 2 \ - --output-path "$DIST_GRAPHBOLT_PATH" \ - --ssh-port 2222 \ - --process-group-timeout 3600 - echo "********* GraphBolt graph construction and partitioning tests passed *********"