Review comment

awslabs · Nov 10, 2023 · a33959d · a33959d
1 parent 681cc09
commit a33959d
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/graphstorm-processing/graphstorm_processing/graph_loaders/dist_heterogeneous_loader.py b/graphstorm-processing/graphstorm_processing/graph_loaders/dist_heterogeneous_loader.py
@@ -161,7 +161,10 @@ def process_and_write_graph_data(
         process_start_time = perf_counter()
 
         if not self._at_least_one_label_exists(data_configs):
-            logging.warning("No labels exist in the dataset, will not produce any masks")
+            logging.warning(
+                "No labels exist in the dataset, will not produce any masks, "
+                "and set task to 'link_prediction'."
+            )
             self.skip_train_masks = True
 
         metadata_dict = self._initialize_metadata_dict(data_configs)