From 60c2239210343ecd7f0c7ddb8dc4b370e2bbff43 Mon Sep 17 00:00:00 2001
From: Rishabh Garg <56191449+RishabhGarg108@users.noreply.github.com>
Date: Tue, 22 Jun 2021 01:09:55 +0530
Subject: [PATCH] Fixed tests from the crazy merge that unexpectedly happened

---
 src/mlpack/tests/decision_tree_test.cpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/tests/decision_tree_test.cpp b/src/mlpack/tests/decision_tree_test.cpp
index e5ad32a03b..1785e05c41 100644
--- a/src/mlpack/tests/decision_tree_test.cpp
+++ b/src/mlpack/tests/decision_tree_test.cpp
@@ -395,7 +395,7 @@ TEST_CASE("RandomBinaryNumericSplitMinSamplesTest", "[DecisionTreeTest]")
   // This should make no difference because it won't split at all.
   const double weightedGain =
       RandomBinaryNumericSplit<GiniGain>::SplitIfBetter<true>(bestGain, values,
-      labels, 2, weights, 8, 1e-7, classProbabilities[0], aux);
+      labels, 2, weights, 8, 1e-7, classProbabilities, aux);
 
   // Make sure that no split was made.
   REQUIRE(gain == DBL_MAX);
@@ -419,13 +419,13 @@ TEST_CASE("RandomBinaryNumericSplitNoGainTest", "[DecisionTreeTest]")
     labels[i + 1] = 1;
   }
 
-  arma::vec classProbabilities(1);
+  arma::vec classProbabilities;
   RandomBinaryNumericSplit<GiniGain>::AuxiliarySplitInfo aux;
 
   // Call the method to do the splitting.
   const double bestGain = GiniGain::Evaluate<false>(labels, 2, weights);
   const double gain = RandomBinaryNumericSplit<GiniGain>::SplitIfBetter<false>(
-      bestGain, values, labels, 2, weights, 10, 1e-7, classProbabilities[0],
+      bestGain, values, labels, 2, weights, 10, 1e-7, classProbabilities,
       aux, true);
 
   // Make sure there was no split.
@@ -449,7 +449,7 @@ TEST_CASE("RandomBinaryNumericSplitDiffSplitTest", "[DecisionTreeTest]")
     labels[i + 1] = 1;
   }
 
-  arma::vec classProbabilities(1), classProbabilities1(1);
+  arma::vec classProbabilities, classProbabilities1;
   BestBinaryNumericSplit<GiniGain>::AuxiliarySplitInfo aux;
   RandomBinaryNumericSplit<GiniGain>::AuxiliarySplitInfo aux1;