From cd40f6564e5ffb81263de6afd4a531b84ad7eeba Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com>
Date: Wed, 17 Jun 2020 11:45:05 -0400
Subject: [PATCH] Add header and fix command (#5082)

---
 examples/adversarial/README.md   | 2 +-
 examples/adversarial/run_hans.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/adversarial/README.md b/examples/adversarial/README.md
index 5d50c84734..3e331a05f4 100644
--- a/examples/adversarial/README.md
+++ b/examples/adversarial/README.md
@@ -11,7 +11,7 @@ export HANS_DIR=path-to-hans
 export MODEL_TYPE=type-of-the-model-e.g.-bert-roberta-xlnet-etc
 export MODEL_PATH=path-to-the-model-directory-that-is-trained-on-NLI-e.g.-by-using-run_glue.py
 
-python examples/adversarial/test_hans.py \
+python run_hans.py \
         --task_name hans \
         --model_type $MODEL_TYPE \
         --do_eval \
diff --git a/examples/adversarial/run_hans.py b/examples/adversarial/run_hans.py
index 00837c12b7..fe384d0727 100644
--- a/examples/adversarial/run_hans.py
+++ b/examples/adversarial/run_hans.py
@@ -216,6 +216,7 @@ def main():
         output_eval_file = os.path.join(training_args.output_dir, "hans_predictions.txt")
         if trainer.is_world_master():
             with open(output_eval_file, "w") as writer:
+                writer.write("pairID,gold_label\n")
                 for pid, pred in zip(pair_ids, preds):
                     writer.write("ex" + str(pid) + "," + label_list[int(pred)] + "\n")