From f1d044eb10f31ad606dae3e81419e1e9d32abcd5 Mon Sep 17 00:00:00 2001
From: Hrayr Muradyan <hrayrmuradyan20@gmail.com>
Date: Mon, 3 Feb 2025 21:51:49 -0800
Subject: [PATCH] fix: fixed/addressed the correlation plot docstring

---
 src/summarease/summarize_numeric.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/summarease/summarize_numeric.py b/src/summarease/summarize_numeric.py
index a2c6996..f099daa 100644
--- a/src/summarease/summarize_numeric.py
+++ b/src/summarease/summarize_numeric.py
@@ -64,10 +64,10 @@ def plot_correlation_heatmap(dataset_numeric: pd.DataFrame):
     save_path : str, optional
         File path to save the generated heatmap. If None, the plot will not be saved.
 
-    Returns:
+    Returns
     -------
-    None
-        Displays the correlation heatmap or optionally saves it to the specified location.
+    alt.Chart
+        The Altair chart visualizing the correlation heatmap.
 
     Example:
     -------