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

fix format

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

4
textattack/dataset_args.py
View File

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