fixes

utils/similarity/encode.py

@@ -22,6 +22,9 @@ slither_logger = logging.getLogger("Slither")
 slither_logger.setLevel(logging.CRITICAL)
 
 def parse_target(target):
+    if target is None:
+        return None, None
+
     parts = target.split('.')
     if len(parts) == 1:
         return None, parts[0]

utils/similarity/info.py

@@ -40,11 +40,11 @@ def info(args):
         x = (filename,contract,fname)
         y = " ".join(irs[x])
 
-        print("Function {} in contract {} is encoded as:".format(fname, contract))
+        logger.info("Function {} in contract {} is encoded as:".format(fname, contract))
-        print(y)
+        logger.info(y)
         if model is not None:
             fvector = model.get_sentence_vector(y)
-            print(fvector)
+            logger.info(fvector)
 
     except Exception:
         logger.error('Error in %s' % args.filename)

utils/similarity/plot.py

@@ -61,7 +61,7 @@ def plot(args):
         tdata = pca.fit_transform(data)
 
         logger.info('Plotting data..')
-        plt.figure()
+        plt.figure(figsize=(20,10))
         assert(len(tdata) == len(fs))
         for ([x,y],l) in zip(tdata, fs):
             x = random.gauss(0, 0.01) + x