mirror of
https://github.com/QData/TextAttack.git
synced 2021-10-13 00:05:06 +03:00
entailment works; add peek-dataset command
This commit is contained in:
@@ -333,6 +333,8 @@ def parse_dataset_from_args(args):
|
||||
dataset_args = args.dataset_from_nlp
|
||||
if ":" in dataset_args:
|
||||
dataset_args = dataset_args.split(":")
|
||||
else:
|
||||
dataset_args = (dataset_args,)
|
||||
dataset = textattack.datasets.HuggingFaceNLPDataset(
|
||||
*dataset_args, shuffle=args.shuffle
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user