update comments

awslabs · Nov 21, 2023 · b40a3c0 · b40a3c0
1 parent 4416e4c
commit b40a3c0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/graphstorm/gconstruct/remap_result.py b/python/graphstorm/gconstruct/remap_result.py
@@ -93,7 +93,7 @@ def write_data_csv_file(data, file_prefix, delimiter=",", col_name_map=None):
         .. code::
 
             nide, emb
-            0, 0.001;1.2000;0.736;
+            0, 0.001;1.2000;0.736;...
 
         Parameters
         ----------
@@ -102,7 +102,7 @@ def write_data_csv_file(data, file_prefix, delimiter=",", col_name_map=None):
         file_prefix: str
             File prefix. The output will be <file_prefix>.parquet.
         delimiter: str
-            Delimiter used to separate file.s
+            Delimiter used to separate columns.
         col_name_map: dict
             A mapping from builtin column name to user defined column name.
     """
@@ -777,6 +777,7 @@ def main(args, gs_config_args):
     col_name_map = None
     if args.column_names is not None:
         col_name_map = {}
+        # Load customized column names
         for col_rename_pair in args.column_names:
             orig_name, new_name = col_rename_pair.split(":")
             assert orig_name in GS_REMAP_BUILTIN_COLS, \