From 8d3a7d7799db1ed040e668f91c17be73aabaab7d Mon Sep 17 00:00:00 2001
From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com>
Date: Tue, 6 Apr 2021 00:51:45 +0530
Subject: [PATCH] Added forgotten import and fixed test

---
 src/mlpack/tests/decision_tree_test.cpp | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src/mlpack/tests/decision_tree_test.cpp b/src/mlpack/tests/decision_tree_test.cpp
index d14bc5fc3a..8aaa7194e2 100644
--- a/src/mlpack/tests/decision_tree_test.cpp
+++ b/src/mlpack/tests/decision_tree_test.cpp
@@ -13,6 +13,7 @@
 #include <mlpack/methods/decision_tree/decision_tree.hpp>
 #include <mlpack/methods/decision_tree/information_gain.hpp>
 #include <mlpack/methods/decision_tree/gini_gain.hpp>
+#include <mlpack/methods/decision_tree/mad_gain.hpp>
 #include <mlpack/methods/decision_tree/random_dimension_select.hpp>
 #include <mlpack/methods/decision_tree/multiple_random_dimension_select.hpp>
 
@@ -47,9 +48,9 @@ TEST_CASE("MADGainNormalTest", "[DecisionTreeRegressionTest")
   arma::rowvec labels(10, arma::fill::randn); // Mean = 0.
 
   // Theoretical gain.
-  const double theoreticalGain = 0.0;
+  double theoreticalGain = 0.0;
   for (size_t i = 0; i < labels.n_elem; ++i)
-    theoreticalGain += std::abs(labels[i]);
+    theoreticalGain -= std::abs(labels[i]);
   theoreticalGain /= (double) labels.n_elem;
 
   // Calculated gain.