download_config.py 635 B

1234567891011121314151617181920212223
  1. # Copyright (c) Alibaba, Inc. and its affiliates.
  2. from typing import Optional, Union
  3. from datasets.download.download_config import DownloadConfig
  4. class DataDownloadConfig(DownloadConfig):
  5. """
  6. Extends `DownloadConfig` with additional attributes for data download.
  7. """
  8. dataset_name: Optional[str] = None
  9. namespace: Optional[str] = None
  10. version: Optional[str] = None
  11. split: Optional[Union[str, list]] = None
  12. data_dir: Optional[str] = None
  13. oss_config: Optional[dict] = {}
  14. meta_args_map: Optional[dict] = {}
  15. num_proc: int = 4
  16. def copy(self) -> 'DataDownloadConfig':
  17. return self