From a3fc8d9fe6c5cdec2fa92f0ec6445242e2de6f55 Mon Sep 17 00:00:00 2001
From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com>
Date: Tue, 11 May 2021 10:02:47 +0530
Subject: [PATCH] Add to CMakeLists.txt

---
 src/mlpack/methods/decision_tree/CMakeLists.txt | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/mlpack/methods/decision_tree/CMakeLists.txt b/src/mlpack/methods/decision_tree/CMakeLists.txt
index 4072a5097d..d36b3d3681 100644
--- a/src/mlpack/methods/decision_tree/CMakeLists.txt
+++ b/src/mlpack/methods/decision_tree/CMakeLists.txt
@@ -4,6 +4,8 @@ set(SOURCES
   all_dimension_select.hpp
   decision_tree.hpp
   decision_tree_impl.hpp
+  decision_tree_regressor.hpp
+  decision_tree_regressor_impl.hpp
   all_categorical_split.hpp
   all_categorical_split_impl.hpp
   best_binary_numeric_split.hpp