From 86f647ca937380cf2ea2569ba5735dcdcc659730 Mon Sep 17 00:00:00 2001
From: Marcus Edel <marcus.edel@fu-berlin.de>
Date: Fri, 24 Apr 2015 00:01:06 +0200
Subject: [PATCH] Return MatType instead of arma::mat.

---
 src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp b/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp
index 95189db95e..a62988e568 100644
--- a/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp
+++ b/src/mlpack/core/tree/rectangle_tree/rectangle_tree.hpp
@@ -303,9 +303,9 @@ class RectangleTree
   RectangleTree*& Parent() { return parent; }
 
   //! Get the dataset which the tree is built on.
-  const arma::mat& Dataset() const { return dataset; }
+  const MatType& Dataset() const { return dataset; }
   //! Modify the dataset which the tree is built on.  Be careful!
-  arma::mat& Dataset() { return const_cast<arma::mat&>(dataset); }
+  MatType& Dataset() { return const_cast<MatType&>(dataset); }
 
   //! Get the points vector for this node.
   const std::vector<size_t>& Points() const { return points; }
@@ -313,9 +313,9 @@ class RectangleTree
   std::vector<size_t>& Points() { return points; }
 
   //! Get the local dataset of this node.
-  const arma::mat& LocalDataset() const { return *localDataset; }
+  const MatType& LocalDataset() const { return *localDataset; }
   //! Modify the local dataset of this node.
-  arma::mat& LocalDataset() { return *localDataset; }
+  MatType& LocalDataset() { return *localDataset; }
 
   //! Get the metric which the tree uses.
   typename HRectBound<>::MetricType Metric() const { return bound.Metric(); }