1 files changed, 4 insertions, 0 deletions
diff --git a/lib/functions.py b/lib/functions.py
index 43ee58a..3630215 100644
--- a/lib/functions.py
+++ b/lib/functions.py
@@ -469,6 +469,10 @@ class CARTFunction(SKLearnRegressionFunction):
         return self.regressor.get_depth()
 
 
+class LMTFunction(SKLearnRegressionFunction):
+    pass
+
+
 class XGBoostFunction(SKLearnRegressionFunction):
     def get_number_of_nodes(self):
         import json