From 98f49beb6d141ac09936e5af0ff5e4a83d09061d Mon Sep 17 00:00:00 2001 From: JalenCato Date: Wed, 25 Oct 2023 22:48:16 +0000 Subject: [PATCH] add doc string --- python/graphstorm/gconstruct/construct_graph.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/python/graphstorm/gconstruct/construct_graph.py b/python/graphstorm/gconstruct/construct_graph.py index 0bfdc66acc..713e19455d 100644 --- a/python/graphstorm/gconstruct/construct_graph.py +++ b/python/graphstorm/gconstruct/construct_graph.py @@ -156,6 +156,8 @@ def parse_edge_data(in_file, feat_ops, label_ops, node_id_map, read_file, The configuration for parsing edge data. skip_nonexist_edges : bool Whether or not to skip edges that don't exist. + ext_mem: str + The address of external memory for multi-column feature Returns ------- @@ -270,6 +272,8 @@ def process_node_data(process_confs, arr_merger, remap_id, ext_mem, num_processe Whether or not to remap node IDs num_processes: int The number of processes to process the input files. + ext_mem: str + The address of external memory for multi-column feature Returns ------- @@ -442,6 +446,8 @@ def process_edge_data(process_confs, node_id_map, arr_merger, The number of processes to process the input files. skip_nonexist_edges : bool Whether or not to skip edges that don't exist. + ext_mem: str + The address of external memory for multi-column feature Returns -------