From 83857ffeaad96a2caa5d3b9b16acbff8260b3a80 Mon Sep 17 00:00:00 2001
From: tseretelitornike <tsereteli.tornike@gmail.com>
Date: Fri, 15 Mar 2019 12:45:48 +0100
Subject: [PATCH] Added missing imports.

---
 examples/run_swag.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/examples/run_swag.py b/examples/run_swag.py
index 2ebd637223..ab8e147cae 100644
--- a/examples/run_swag.py
+++ b/examples/run_swag.py
@@ -15,6 +15,8 @@
 # limitations under the License.
 """BERT finetuning runner."""
 
+from __future__ import absolute_import
+
 import argparse
 import csv
 import logging
@@ -31,7 +33,7 @@ from torch.utils.data.distributed import DistributedSampler
 from tqdm import tqdm, trange
 
 from pytorch_pretrained_bert.file_utils import PYTORCH_PRETRAINED_BERT_CACHE
-from pytorch_pretrained_bert.modeling import BertForMultipleChoice
+from pytorch_pretrained_bert.modeling import (BertForMultipleChoice, BertConfig, WEIGHTS_NAME, CONFIG_NAME)
 from pytorch_pretrained_bert.optimization import BertAdam, warmup_linear
 from pytorch_pretrained_bert.tokenization import BertTokenizer