Merge pull request #533 from QData/fix-dataset-split-bug

Fix dataset-split bug
QData · Oct 5, 2021 · caacc1c · caacc1c
2 parents 0f6401d + 9ba7d9c
commit caacc1c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/textattack/dataset_args.py b/textattack/dataset_args.py
@@ -275,7 +275,9 @@ def _create_dataset_from_args(cls, args):
                     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
                     )