Now passes style guide enforcement

This commit is contained in:
Martin Malmsten committed 2020-02-23 21:47:59 +01:00
1 parent 33eb8a165d
commit 105dcb4162
1 file changed
+1 -1
+1 -1
View File
@@ -612,7 +612,7 @@ def main():
label2id={label: i for i, label in enumerate(labels)},
cache_dir=args.cache_dir if args.cache_dir else None,
)
tokenizer_args = {k: v for k, v in vars(args).items() if v != None and k in TOKENIZER_ARGS}
tokenizer_args = {k: v for k, v in vars(args).items() if v is not None and k in TOKENIZER_ARGS}
logger.info("Tokenizer arguments: %s", tokenizer_args)
tokenizer = tokenizer_class.from_pretrained(
args.tokenizer_name if args.tokenizer_name else args.model_name_or_path,