Fix E265 flake8 warning (x1).

This commit is contained in:
Aymeric Augustin committed 2019-12-22 10:59:07 +01:00
1 parent 28e608a2c2
commit f9c5317db2
1 file changed
+1 -1
+1 -1
View File
@@ -356,7 +356,7 @@ def train(args, train_dataset, model, tokenizer):
inputs = {
"input_ids": batch[0],
"attention_mask": batch[1],
#'token_type_ids': None if args.model_type == 'xlm' else batch[2],
# 'token_type_ids': None if args.model_type == 'xlm' else batch[2],
"token_type_ids": batch[2],
"labels": batch[3],
}