Skip to content
Snippets Groups Projects
Commit 4228616a authored by Marcin Wątroba's avatar Marcin Wątroba
Browse files

download_dataset command

parent 1936693a
Branches
No related merge requests found
......@@ -5,7 +5,7 @@ import datasets
def download_dataset(dataset_path: str, dataset_name: Optional[str], cache_dir: str):
dataset = datasets.load_dataset(dataset_path, dataset_name, cache_dir=cache_dir)
dataset = datasets.load_dataset(dataset_path, dataset_name if len(dataset_name) > 0 else None, cache_dir=cache_dir)
print(dataset)
......@@ -15,4 +15,4 @@ if __name__ == '__main__':
parser.add_argument("--dataset_name")
parser.add_argument("--cache_dir")
args = parser.parse_args()
download_dataset(args.dataset, args.asr)
download_dataset(args.dataset_path, args.dataset_name, args.cache_dir)
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment