From 2df938046637dfd8c53e27118b1047134470d206 Mon Sep 17 00:00:00 2001 From: Xiang Song Date: Wed, 24 Apr 2024 12:18:21 -0700 Subject: [PATCH] Update error string --- python/graphstorm/gconstruct/transform.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/python/graphstorm/gconstruct/transform.py b/python/graphstorm/gconstruct/transform.py index b18a3652b8..bde4a63f93 100644 --- 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