diff --git a/toolkit.py b/toolkit.py index 09fcacd840250896811b076ee5fdce32f3cb05fc..50cfa16fd9ff1f434429fd4f61431e484d4e1c22 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