From 58e12fb34304b9001dd3516f7555b5e0853b9b4e Mon Sep 17 00:00:00 2001 From: Theodore Vasiloudis Date: Wed, 12 Jun 2024 18:38:31 +0000 Subject: [PATCH] Add default value for DGL tools in dist part --- python/graphstorm/gpartition/dist_partition_graph.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/graphstorm/gpartition/dist_partition_graph.py b/python/graphstorm/gpartition/dist_partition_graph.py index 17fae41161..9040d25879 100644 --- a/python/graphstorm/gpartition/dist_partition_graph.py +++ b/python/graphstorm/gpartition/dist_partition_graph.py @@ -163,7 +163,7 @@ def parse_args() -> argparse.Namespace: argparser.add_argument("--num-parts", type=int, required=True, help="Number of partitions to generate") argparser.add_argument("--ssh-port", type=int, default=22, help="SSH Port") - argparser.add_argument("--dgl-tool-path", type=str, + argparser.add_argument("--dgl-tool-path", type=str, default="/root/dgl/tools", help="The path to dgl/tools") argparser.add_argument("--partition-algorithm", type=str, default="random", choices=["random", "parmetis"], help="Partition algorithm to use.")