diff --git a/utils/fed_util.py b/utils/fed_util.py
index e462407..fea2bc5 100644
--- a/utils/fed_util.py
+++ b/utils/fed_util.py
@@ -25,6 +25,10 @@ def _parse_yolo_label_file(label_path: str) -> Set[int]:
     Return a set of class_ids found in a YOLO .txt label file.
     Empty file -> empty set. Missing file -> empty set.
     Robust to blank lines / trailing spaces.
+    Args:
+        label_path: path to the label file
+    Returns:
+        set of class IDs (integers) found in the file
     """
     class_ids: Set[int] = set()
     if not os.path.exists(label_path):