alihan/textattack-nlp-transformer
1
0
Fork 0
mirror of https://github.com/QData/TextAttack.git synced 2021-10-13 00:05:06 +03:00
Code Issues Projects Releases Wiki Activity

Update dataset_args.py

Browse Source
This commit is contained in:
Hanyu Liu
2021-09-30 23:11:04 -04:00
parent 7d2de08771
commit 014af17184
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/dataset_args.py
View File

@@ -275,7 +275,7 @@ class DatasetArgs:
dataset_args = (dataset_args,)
if args.dataset_split:
if len(dataset_args) > 1:
dataset_args[2] = args.dataset_split
dataset_args = dataset_args[:1] + (args.dataset_split,) + dataset_args[2:]
dataset = textattack.datasets.HuggingFaceDataset(
*dataset_args, shuffle=False
)