Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-22 06:08:46 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix docstring

Browse source
This commit is contained in:
Benjamin Trom 2024-02-07 16:08:46 +01:00
parent 2e526d35d1
commit bbe0759151
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/refiners/training_utils/trainer.py
View file

@ -252,7 +252,7 @@ ConfigType = TypeVar("ConfigType", bound=BaseConfig)
class _Dataset(Dataset[Batch]): class _Dataset(Dataset[Batch]):
""" """
A wrapper around the `get_item` method to create a [`torch.utils.data.Dataset`] A wrapper around the `get_item` method to create a [`torch.utils.data.Dataset`][torch.utils.data.Dataset].
""" """
def __init__(self, get_item: Callable[[int], Batch], length: int) -> None: def __init__(self, get_item: Callable[[int], Batch], length: int) -> None: