Upload loaders.py with huggingface_hub
Browse files- loaders.py +2 -1
loaders.py
CHANGED
@@ -26,11 +26,12 @@ class LoadHF(Loader):
|
|
26 |
name: Optional[str] = None
|
27 |
data_dir: Optional[str] = None
|
28 |
data_files: Optional[Union[str, Sequence[str], Mapping[str, Union[str, Sequence[str]]]]] = None
|
|
|
29 |
cached = False
|
30 |
|
31 |
def process(self):
|
32 |
dataset = hf_load_dataset(
|
33 |
-
self.path, name=self.name, data_dir=self.data_dir, data_files=self.data_files, streaming=
|
34 |
)
|
35 |
|
36 |
return MultiStream.from_iterables(dataset)
|
|
|
26 |
name: Optional[str] = None
|
27 |
data_dir: Optional[str] = None
|
28 |
data_files: Optional[Union[str, Sequence[str], Mapping[str, Union[str, Sequence[str]]]]] = None
|
29 |
+
streaming: bool = True
|
30 |
cached = False
|
31 |
|
32 |
def process(self):
|
33 |
dataset = hf_load_dataset(
|
34 |
+
self.path, name=self.name, data_dir=self.data_dir, data_files=self.data_files, streaming=self.streaming
|
35 |
)
|
36 |
|
37 |
return MultiStream.from_iterables(dataset)
|