Update error string

awslabs · Apr 24, 2024 · 2df9380 · 2df9380
1 parent 4d9debf
commit 2df9380
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py
@@ -1397,7 +1397,9 @@ def __init__(self, col_name, label_name, id_col, task_type,
         self._stats_type = stats_type
 
         assert isinstance(mask_field_names, tuple) and len(mask_field_names) == 3, \
-            "mask_field_names must be a tuple with three strings."
+            "mask_field_names must be a tuple with three strings " \
+            "for training mask, validation mask and test mask, respectively." \
+            "For example ('tmask', 'vmask', 'tmask')."
         self._mask_field_names = mask_field_names
 
     @property
@@ -1528,7 +1530,9 @@ def __init__(self, col_name, label_name, split_pct,
         self._split_pct = split_pct
         self._stats_type = stats_type
         assert isinstance(mask_field_names, tuple) and len(mask_field_names) == 3, \
-            "mask_field_names must be a tuple with three strings."
+            "mask_field_names must be a tuple with three strings " \
+            "for training mask, validation mask and test mask, respectively." \
+            "For example ('tmask', 'vmask', 'tmask')."
         self._mask_field_names = mask_field_names
 
     @property