summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/functions.py1
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/functions.py b/lib/functions.py
index ac92af9..5aa773b 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -539,6 +539,7 @@ class LMTFunction(SKLearnRegressionFunction):
return self.regressor.node_count
def get_max_depth(self):
+ # FIXME this returns the configured maximum depth, not the actual tree depth.
return self.regressor.max_depth