Update

awslabs · Apr 28, 2024 · 02461f6 · 02461f6
1 parent 71909b7
commit 02461f6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 6 deletions.
diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py
@@ -1726,7 +1726,9 @@ def parse_label_ops(confs, is_node):
     """
     label_confs = confs['labels']
     assert len(label_confs) >= 1, \
-        "There must be at least one label configuration under the 'labels' config block."
+        "If a 'labels' field is defined for a node type or an edge type in the " \
+        "configuration file, it should not be empty."
+
     def parse_label_conf(label_conf):
         assert 'task_type' in label_conf, "'task_type' must be defined in the label field."
         task_type = label_conf['task_type']

diff --git a/tests/end2end-tests/data_process/test_multitask_data.py b/tests/end2end-tests/data_process/test_multitask_data.py
@@ -44,11 +44,6 @@ def read_data_parquet(data_file):
 with open(args.conf_file, 'r') as f:
     conf = json.load(f)
 
-args = argparser.parse_args()
-out_dir = args.graph_dir
-with open(args.conf_file, 'r') as f:
-    conf = json.load(f)
-
 if args.graph_format == "DGL":
     g = dgl.load_graphs(os.path.join(out_dir, "test.dgl"))[0][0]
 elif args.graph_format == "DistDGL":