Add default value for DGL tools in dist part

awslabs · Jun 12, 2024 · 58e12fb · 58e12fb
1 parent cac77a8
commit 58e12fb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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.")