diff --git a/examples/contrib/run_swag.py b/examples/contrib/run_swag.py
index 65c07c2a32..a996a2ac9d 100644
--- a/examples/contrib/run_swag.py
+++ b/examples/contrib/run_swag.py
@@ -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],
             }