Improve hf transform docstring

grok-ai · Oct 12, 2023 · df7afb7 · df7afb7
1 parent f5a1884
commit df7afb7
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/utils/hf_io.py b/{{ cookiecutter.repository_name }}/src/{{ cookiecutter.package_name }}/utils/hf_io.py
@@ -18,7 +18,7 @@ def __init__(
  key: str,
  transform: Callable[[torch.Tensor], torch.Tensor],
  ):
- """Apply a transform to a dataset column.
+ """Apply a row-wise transform to a dataset column.
 
  Args:
  key (str): The key of the column to transform.
@@ -90,7 +90,7 @@ def load_hf_dataset(**cfg: DictConfig) -> MetadataDatasetDict:
  It will contain the following splits:
  - train
  - val
- - test.
+ - test
  If `val_split` is not specified in the config, it will be created from the train split
  according to the `val_percentage` specified in the config.