From 1bb1de2bd8b0144174142d17090b27687e7a186c Mon Sep 17 00:00:00 2001 From: moto <855818+mthrok@users.noreply.github.com> Date: Tue, 19 Nov 2024 09:21:21 -0500 Subject: [PATCH] Document `src` attribute of DataLoader (#281) --- src/spdl/dataloader/_dataloader.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/spdl/dataloader/_dataloader.py b/src/spdl/dataloader/_dataloader.py index 427712f4..487e302a 100644 --- a/src/spdl/dataloader/_dataloader.py +++ b/src/spdl/dataloader/_dataloader.py @@ -177,6 +177,8 @@ class DataLoader(Generic[Source, Output]): - :py:func:`spdl.io.transfer_buffer`: Sending the data to GPU. - :py:func:`spdl.io.to_torch`, :py:func:`spdl.io.to_numba`, :py:func:`spdl.io.to_jax`: Casting the memroy buffer to array type. + + :ivar src: The source object provided in the constructor. """ def __init__(