relative import in compare

compare.py

@@ -7,7 +7,7 @@ import numpy as np
 import matplotlib.pyplot as plt
 from sklearn.metrics import roc_curve, auc
 
-from toolkit import ClassificationProject
+from .toolkit import ClassificationProject
 
 """
 A few functions to compare different setups

toolkit.py

@@ -764,6 +764,8 @@ if __name__ == "__main__":
 
     filename = "/project/etp4/nhartmann/trees/allTrees_m1.8_NoSys.root"
 
+
+
     c = ClassificationProject("test4",
                               signal_trees = [(filename, "GG_oneStep_1705_1105_505_NoSys")],
                               bkg_trees = [(filename, "ttbar_NoSys"),