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

raise warning (not error) when too few samples to attack

Browse Source
This commit is contained in:
Jack Morris
2020-07-07 22:08:51 -04:00
parent 7e3b90e979
commit 58609feac1
2 changed files with 10 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
textattack/commands/attack/attack_args_helpers.py
View File

@@ -428,7 +428,7 @@ def parse_logger_from_args(args):
color_method = None if args.enable_csv == "plain" else "file"
csv_path = os.path.join(args.out_dir, outfile_name)
attack_log_manager.add_output_csv(csv_path, color_method)
print("Logging to CSV at path {}.".format(csv_path))
textattack.shared.logger.info(f"Logging to CSV at path {csv_path}.")
# Visdom
if args.enable_visdom: