Elron commited on
Commit
2e01f35
1 Parent(s): 2b8594d

Upload loaders.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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=True
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)