Merge pull request #3 from avinashsai/master

Fix import error
facebookresearch · Oct 27, 2020 · f2f0b68 · f2f0b68
2 parents 6c7bff2 + 13674ed
commit f2f0b68
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/flat_retrieve.py b/src/flat_retrieve.py
@@ -7,6 +7,7 @@
 """
 
 import os
+import sys
 import torch
 import argparse
 import time