Merge pull request #378 from huggingface/absolute_imports

Add absolute imports to GPT, GPT-2, Transfo-XL and and fix empty nbest_predictions.json

examples/run_squad.py

@@ -620,7 +620,7 @@ def write_predictions(all_examples, all_features, all_results, n_best_size,
                 all_predictions[example.qas_id] = ""
             else:
                 all_predictions[example.qas_id] = best_non_null_entry.text
-                all_nbest_json[example.qas_id] = nbest_json
+            all_nbest_json[example.qas_id] = nbest_json
 
     with open(output_prediction_file, "w") as writer:
         writer.write(json.dumps(all_predictions, indent=4) + "\n")

pytorch_pretrained_bert/modeling_gpt2.py

@@ -15,6 +15,8 @@
 # limitations under the License.
 """PyTorch OpenAI GPT-2 model."""
 
+from __future__ import absolute_import, division, print_function, unicode_literals
+
 import collections
 import copy
 import json

pytorch_pretrained_bert/modeling_openai.py

@@ -15,6 +15,8 @@
 # limitations under the License.
 """PyTorch OpenAI GPT model."""
 
+from __future__ import absolute_import, division, print_function, unicode_literals
+
 import collections
 import copy
 import json

pytorch_pretrained_bert/modeling_transfo_xl.py

@@ -18,6 +18,8 @@
     In particular https://github.com/kimiyoung/transformer-xl/blob/master/pytorch/mem_transformer.py
 """
 
+from __future__ import absolute_import, division, print_function, unicode_literals
+
 import os
 import copy
 import json