From 74343a15e17723cc89df3d757d10dec777061452 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ricardo=20Monta=C3=B1ana?= <rmontanana@gmail.com>
Date: Wed, 24 Nov 2021 10:50:19 +0100
Subject: [PATCH] Fix nodes_leaves for base_estimator

---
 odte/Odte.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/odte/Odte.py b/odte/Odte.py
index 890f649..e1526b9 100644
--- a/odte/Odte.py
+++ b/odte/Odte.py
@@ -88,13 +88,14 @@ class Odte(BaseEnsemble, ClassifierMixin):
         return self
 
     def _compute_metrics(self) -> None:
-        tdepth = tnodes = tleaves = 0
+        tdepth = tnodes = tleaves = 0.0
         for estimator in self.estimators_:
-            nodes, leaves = estimator.nodes_leaves()
-            depth = estimator.depth_
-            tdepth += depth
-            tnodes += nodes
-            tleaves += leaves
+            if hasattr(estimator, "nodes_leaves"):
+                nodes, leaves = estimator.nodes_leaves()
+                depth = estimator.depth_
+                tdepth += depth
+                tnodes += nodes
+                tleaves += leaves
         self.depth_ = tdepth / self.n_estimators
         self.leaves_ = tleaves / self.n_estimators
         self.nodes_ = tnodes / self.n_estimators