From 13674eddd8d367bf79692610cb1b68f3580117c0 Mon Sep 17 00:00:00 2001
From: Avinash Madasu <avinash.sai001@gmail.com>
Date: Tue, 27 Oct 2020 13:55:30 +0530
Subject: [PATCH] Fix import error

---
 src/flat_retrieve.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/flat_retrieve.py b/src/flat_retrieve.py
index ff96eac..bc75e7d 100644
--- a/src/flat_retrieve.py
+++ b/src/flat_retrieve.py
@@ -7,6 +7,7 @@
 """
 
 import os
+import sys
 import torch
 import argparse
 import time