Fix import on initalization

run_classifier_pytorch.py

@@ -23,6 +23,7 @@ import os
 import logging
 import argparse
 
+import random
 import numpy as np
 from tqdm import tqdm, trange
 import torch

run_squad_pytorch.py

@@ -26,6 +26,7 @@ import json
 import math
 import os
 from tqdm import tqdm, trange
+import random
 
 import torch
 from torch.utils.data import TensorDataset, DataLoader, RandomSampler, SequentialSampler