From 85b9fde9a78c6bc3c9b8e2899240fc77a7d5adf4 Mon Sep 17 00:00:00 2001
From: Malte Londschien <malte@londschien.de>
Date: Wed, 20 Oct 2021 17:04:24 +0200
Subject: [PATCH] Another format.

---
 src/ensemble/random_forest_regressor.rs | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/src/ensemble/random_forest_regressor.rs b/src/ensemble/random_forest_regressor.rs
index f1caa8e..90ac479 100644
--- a/src/ensemble/random_forest_regressor.rs
+++ b/src/ensemble/random_forest_regressor.rs
@@ -203,7 +203,11 @@ impl<T: RealNumber> RandomForestRegressor<T> {
             trees.push(tree);
         }
 
-        Ok(RandomForestRegressor { parameters, trees, samples: maybe_all_samples })
+        Ok(RandomForestRegressor {
+            parameters,
+            trees,
+            samples: maybe_all_samples,
+        })
     }
 
     /// Predict class for `x`
@@ -232,7 +236,6 @@ impl<T: RealNumber> RandomForestRegressor<T> {
         result / T::from(n_trees).unwrap()
     }
 
-
     /// Predict OOB classes for `x`. `x` is expected to be equal to the dataset used in training.
     pub fn predict_oob<M: Matrix<T>>(&self, x: &M) -> Result<M::RowVector, Failed> {
         let (n, _) = x.shape();
@@ -370,7 +373,8 @@ mod tests {
                 m: Option::None,
                 keep_samples: true,
             },
-        ).unwrap();
+        )
+        .unwrap();
 
         let y_hat = regressor.predict(&x).unwrap();
         let y_hat_oob = regressor.predict_oob(&x).unwrap();