From 5a551178236410d6dc669863cbf3d19aadb9bd51 Mon Sep 17 00:00:00 2001 From: Thomas Wolf Date: Tue, 31 Oct 2023 11:12:53 +0000 Subject: [PATCH] fix lint --- src/datatrove/io/base.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/datatrove/io/base.py b/src/datatrove/io/base.py index 8f8e8b87..93f36cb2 100644 --- a/src/datatrove/io/base.py +++ b/src/datatrove/io/base.py @@ -66,14 +66,15 @@ def open(self, binary=False, compression: Literal["gzip", "zst"] | None = None): @dataclass class BaseInputDataFolder(ABC): - """ An input data folder - - Args: + """An input data folder + + Args: path (str): path to the folder extension (str | list[str], optional): file extensions to filter. Defaults to None. recursive (bool, optional): whether to search recursively. Defaults to True. match_pattern (str, optional): pattern to match file names. Defaults to None. """ + path: str extension: str | list[str] = None recursive: bool = True