From 212e886c2030844e0ceff675bfb93829a118d3f8 Mon Sep 17 00:00:00 2001
From: Nikolai Hartmann <Nikolai.Hartmann@physik.uni-muenchen.de>
Date: Tue, 28 Aug 2018 14:16:49 +0200
Subject: [PATCH] fix initialisation of ClassificationProjectDataFrame
---
toolkit.py | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/toolkit.py b/toolkit.py
index 09fcacd..50cfa16 100755
--- a/toolkit.py
+++ b/toolkit.py
@@ -1523,9 +1523,11 @@ class ClassificationProjectDataFrame(ClassificationProject):
raise NotImplementedError("'split_column' is the only currently supported split mode")
self.split_mode = split_mode
self.split_column = split_column
- super(ClassificationProjectDataFrame, self).__init__(name,
- signal_trees=[], bkg_trees=[], branches=[], weight_expr="1",
- **kwargs)
+ super(ClassificationProjectDataFrame, self)._init_from_args(
+ name,
+ signal_trees=[], bkg_trees=[], branches=[], weight_expr="1",
+ **kwargs
+ )
self._x_train = None
self._x_test = None
self._y_train = None
--
GitLab