Revert "Update builtin train/inference entries to accept unknown argu…

…ments (#681)"

This reverts commit ca0ed1f.

examples/customized_models/HGT/hgt_nc.py

@@ -393,8 +393,7 @@ def main(args):
     argparser.add_argument("--local_rank", type=int,
                            help="The rank of the trainer. \
                                  For customized models, MUST have this argument!!")
+    args = argparser.parse_args()
 
-    # Ignore unknown args to make script more robust to input arguments
-    args, _ = argparser.parse_known_args()
     print(args)
     main(args)

examples/temporal_graph_learning/main_nc.py

@@ -106,8 +106,8 @@ def generate_parser():
 if __name__ == "__main__":
     arg_parser = generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    args, _ = arg_parser.parse_known_args()
+    args = arg_parser.parse_args()
+    print(args)
     main(args)
 
 

python/graphstorm/run/gsgnn_dt/distill_gnn.py

@@ -92,6 +92,6 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    print(args)
+    main(args)

python/graphstorm/run/gsgnn_emb/gsgnn_node_emb.py

@@ -103,6 +103,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser = generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_ep/ep_infer_gnn.py

@@ -99,6 +99,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_ep/ep_infer_lm.py

@@ -89,6 +89,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_ep/gsgnn_ep.py

@@ -179,6 +179,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_ep/gsgnn_lm_ep.py

@@ -148,6 +148,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_lp/gsgnn_lm_lp.py

@@ -197,6 +197,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_lp/gsgnn_lp.py

@@ -223,6 +223,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_lp/lp_infer_gnn.py

@@ -88,6 +88,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_lp/lp_infer_lm.py

@@ -90,6 +90,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_np/gsgnn_np.py

@@ -190,6 +190,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

python/graphstorm/run/gsgnn_np/np_infer_gnn.py

@@ -94,6 +94,5 @@ def generate_parser():
 if __name__ == '__main__':
     arg_parser=generate_parser()
 
-    # Ignore unknown args to make script more robust to input arguments
-    gs_args, _ = arg_parser.parse_known_args()
-    main(gs_args)
+    args = arg_parser.parse_args()
+    main(args)

sagemaker/launch/launch_infer.py

@@ -92,7 +92,6 @@ def run_job(input_args, image, unknownargs):
     # We must handle cases like
     # --target-etype query,clicks,asin query,search,asin
     # --feat-name ntype0:feat0 ntype1:feat1
-    # --column-names nid,~id emb,embedding
     unknow_idx = 0
     while unknow_idx < len(unknownargs):
         print(unknownargs[unknow_idx])