From 1574f97333ff479f14898ada6e55be34355c32d0 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 8 May 2018 14:24:49 +0200
Subject: [PATCH] relative import in compare
---
compare.py | 2 +-
toolkit.py | 2 ++
2 files changed, 3 insertions(+), 1 deletion(-)
diff --git a/compare.py b/compare.py
index f0a2a3d..e6496db 100755
--- a/compare.py
+++ b/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
diff --git a/toolkit.py b/toolkit.py
index 271989d..a0d3a1c 100755
--- a/toolkit.py
+++ b/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"),
--
GitLab