Fix docstring

awslabs · Dec 19, 2024 · b6306c8 · b6306c8
1 parent 27a1025
commit b6306c8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graphstorm-processing/graphstorm_processing/data_transformations/dist_label_loader.py b/graphstorm-processing/graphstorm_processing/data_transformations/dist_label_loader.py
@@ -75,9 +75,9 @@ class CustomSplit:
     ----------
     train : list[str]
         Paths of the training mask parquet files.
-    valid : str
+    valid : list[str]
         Paths of the validation mask parquet files.
-    test : str
+    test : list[str]
         Paths of the testing mask parquet files.
     mask_columns : list[str]
         List of columns that contain original string ids.