Add comments

facebookresearch · Nov 18, 2024 · 5a1d4e7 · 5a1d4e7
1 parent bff4819
commit 5a1d4e7
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/spdl/dataloader/_dataloader.py b/src/spdl/dataloader/_dataloader.py
@@ -183,11 +183,15 @@ def __init__(
         self,
         src: Iterable[Source] | AsyncIterable[Source],
         *,
+        # Pre-processing
         preprocessor: Callables[[Source], T] | None = None,
+        # Aggregation
         batch_size: int | None = None,
         drop_last: bool = False,
         aggregator: Functions[[list[T]], Output] | None = None,
+        # Buffering
         buffer_size: int = 3,
+        # Execution config
         num_threads: int = 8,
         timeout: float | None = None,
         output_order: str = "completion",