From cd09cd5b40fa9ccab1a40643bd5cf1245fcc4a2c Mon Sep 17 00:00:00 2001 From: VictorSanh Date: Sat, 3 Nov 2018 15:38:30 -0400 Subject: [PATCH] Fix import on initalization --- run_classifier_pytorch.py | 1 + run_squad_pytorch.py | 1 + 2 files changed, 2 insertions(+) diff --git a/run_classifier_pytorch.py b/run_classifier_pytorch.py index 9852791163..eaa32e4790 100644 --- a/run_classifier_pytorch.py +++ b/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 diff --git a/run_squad_pytorch.py b/run_squad_pytorch.py index a1db682cd4..701aa229a2 100644 --- a/run_squad_pytorch.py +++ b/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