Fix bug

julianspaeth · Oct 25, 2019 · 4c2616c · 4c2616c
1 parent d9d7a77
commit 4c2616c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/random_survival_forest/Node.py b/random_survival_forest/Node.py
@@ -57,11 +57,11 @@ def grow_tree(self):
 
         lf_idxs, rf_idxs = select_new_feature_indices(self.random_state, self.x, self.n_features)
 
-        self.lhs = Node(self.x.iloc[lhs_idxs_opt, :], self.y.iloc[lhs_idxs_opt, :], self.tree,
-                        lf_idxs, self.n_features, min_leaf=self.min_leaf, random_state=self.random_state)
+        self.lhs = Node(self.x.iloc[lhs_idxs_opt, :], self.y.iloc[lhs_idxs_opt, :], self.tree, lf_idxs,
+                        self.n_features, timeline=self.timeline, min_leaf=self.min_leaf, random_state=self.random_state)
 
-        self.rhs = Node(self.x.iloc[rhs_idxs_opt, :], self.y.iloc[rhs_idxs_opt, :], self.tree,
-                        rf_idxs, self.n_features, min_leaf=self.min_leaf, random_state=self.random_state)
+        self.rhs = Node(self.x.iloc[rhs_idxs_opt, :], self.y.iloc[rhs_idxs_opt, :], self.tree, rf_idxs,
+                        self.n_features, timeline=self.timeline, min_leaf=self.min_leaf, random_state=self.random_state)
 
         return self
 

diff --git a/setup.py b/setup.py
@@ -6,7 +6,7 @@
 setup(
   name='random_survival_forest',         # How you named your package folder (MyLib)
   packages=['random_survival_forest'],   # Chose the same as "name"
-  version='0.6.1',      # Start with a small number and increase it with every change you make
+  version='0.6.2',      # Start with a small number and increase it with every change you make
   license="MIT License",        # Chose a license from here: https://help.github.com/articles/licensing-a-repository
   long_description=readme,
   long_description_content_type="text/markdown",