improved module loading

utils/similarity/__init__.py

@@ -1,9 +1 @@
-# from https://stackoverflow.com/questions/563022/whats-python-good-practice-for-importing-and-offering-optional-features
-import sys
-
-try:
-    import fastText
-except ImportError:
-    print("ERROR: in order to use slither-simil, you need to install fastText 0.2.0:")
-    print("$ pip3 install https://github.com/facebookresearch/fastText/archive/0.2.0.zip --user\n")
-    sys.exit(-1)
+from .model import load_model

utils/similarity/info.py

@@ -3,7 +3,7 @@ import sys
 import os.path
 import traceback
 
-from fastText import load_model
+from .model  import load_model
 from .encode import parse_target, encode_contract
 
 logging.basicConfig()

utils/similarity/model.py

@@ -0,0 +1,11 @@
+import sys
+
+try:
+    from fastText import load_model
+    from fastText import train_unsupervised
+except ImportError:
+    print("ERROR: in order to use slither-simil, you need to install fastText 0.2.0:")
+    print("$ pip3 install https://github.com/facebookresearch/fastText/archive/0.2.0.zip --user\n")
+    sys.exit(-1)
+
+

utils/similarity/plot.py

@@ -5,6 +5,9 @@ import operator
 import numpy as np
 import random
 
+from .model  import load_model   
+from .encode import load_and_encode, parse_target
+
 try:
     from sklearn import decomposition
     import matplotlib.pyplot as plt
@@ -12,9 +15,6 @@ except ImportError:
     decomposition = None
     plt = None
 
-from fastText import load_model
-from .encode import load_and_encode, parse_target
-
 logger = logging.getLogger("Slither-simil")
 
 def plot(args):

utils/similarity/test.py

@@ -5,7 +5,7 @@ import traceback
 import operator
 import numpy as np
 
-from fastText import load_model
+from .model      import load_model
 from .encode     import encode_contract, load_and_encode, parse_target
 from .cache      import save_cache
 from .similarity import similarity

utils/similarity/train.py

@@ -5,7 +5,7 @@ import traceback
 import operator
 import os
 
-from fastText import train_unsupervised
+from .model   import train_unsupervised
 from .encode  import encode_contract, load_contracts
 from .cache   import save_cache