Update graphbolt-graph-construction.sh

awslabs · Dec 19, 2024 · e4d1e21 · e4d1e21
1 parent dae5436
commit e4d1e21
Showing 1 changed file with 2 additions and 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
@@ -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 *********"