use absolute imports

airbnb · Aug 8, 2018 · d2f5b67 · d2f5b67
1 parent 2f0789b
commit d2f5b67
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/Adversary/__init__.py b/Adversary/__init__.py
@@ -1 +1,2 @@
-from Adversary import Adversary
+from Adversary.adversary import Adversary
+from Adversary.attacks import *
diff --git a/Adversary/Adversary.py → Adversary/adversary.py b/Adversary/Adversary.py → Adversary/adversary.py
@@ -5,8 +5,8 @@
 import pandas as pd
 from textblob import TextBlob
 
-from attacks import *
-from utils import *
+from Adversary.attacks import *
+from Adversary.utils import *
 
 
 class Adversary:

diff --git a/Adversary/attacks.py b/Adversary/attacks.py
@@ -1,7 +1,7 @@
 from random import choice, randint, sample, randrange
 from string import punctuation
 
-from constants import *
+from Adversary.constants import *
 
 '''These act on a single text'''
 

diff --git a/tests/test_adversary.py b/tests/test_adversary.py
@@ -1,4 +1,4 @@
-from Adversary import Adversary
+from Adversary.adversary import Adversary
 
 def test_generate_single_iter():
     m = Adversary(verbose=True)