From 079428dec59fb1c39f494d2a08a11690e63a4219 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Thu, 8 Jul 2021 22:06:18 -0500
Subject: [PATCH] Fixing mising import

---
 hpvm/test/epoch_dnn/main.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/hpvm/test/epoch_dnn/main.py b/hpvm/test/epoch_dnn/main.py
index e94b13f48c..2ecfa5343a 100644
--- a/hpvm/test/epoch_dnn/main.py
+++ b/hpvm/test/epoch_dnn/main.py
@@ -2,6 +2,7 @@ import site
 from pathlib import Path
 from os import makedirs
 import numpy as np
+import sys
 
 import torch
 from torch2hpvm import BinDataset, ModelExporter
-- 
GitLab