From 5d58090f407a29c8700c360b423c42a91a8e4fee Mon Sep 17 00:00:00 2001
From: Cristine Guadelupe <cristineguadelupe@me.com>
Date: Fri, 15 Dec 2023 14:45:36 -0300
Subject: [PATCH] Refactor: s_correlation (#774)

---
 native/explorer/src/series.rs | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/native/explorer/src/series.rs b/native/explorer/src/series.rs
index 11deb2b22..cba8955b4 100644
--- a/native/explorer/src/series.rs
+++ b/native/explorer/src/series.rs
@@ -1026,13 +1026,11 @@ pub fn s_correlation(
     let corr = match method {
         ExCorrelationMethod::Pearson => pearson_corr(s1.f64()?, s2.f64()?, ddof),
         ExCorrelationMethod::Spearman => {
-            let df = df!("s1" => s1, "s2" => s2)?;
-            let lazy_df = df
+            let df = df!("s1" => s1, "s2" => s2)?
                 .lazy()
-                .with_column(spearman_rank_corr(col("s1"), col("s2"), ddof, true).alias("corr"));
-            let result = lazy_df.collect()?;
-            let item = result.column("corr")?.get(0)?;
-            match item {
+                .with_column(spearman_rank_corr(col("s1"), col("s2"), ddof, true).alias("corr"))
+                .collect()?;
+            match df.column("corr")?.get(0)? {
                 AnyValue::Float64(x) => Some(x),
                 _ => None,
             }