From 20e16861bde6faa83e43288d1cd31db8858e4e59 Mon Sep 17 00:00:00 2001
From: Caleb Evans <caleb@calebevans.me>
Date: Wed, 21 Feb 2024 16:21:37 -0800
Subject: [PATCH] Simplify column rename when outputting resulting dataframe

---
 ica/core.py | 5 +----
 1 file changed, 1 insertion(+), 4 deletions(-)

diff --git a/ica/core.py b/ica/core.py
index 96a21fb..6353ec0 100644
--- a/ica/core.py
+++ b/ica/core.py
@@ -262,10 +262,7 @@ def prepare_df_for_output(df: pd.DataFrame) -> pd.DataFrame:
             # Prettify header column (i.e. textual values in first column)
             index=prettify_header_name,
             # Prettify header row (i.e. column names)
-            columns={
-                column_name: prettify_header_name(column_name)
-                for column_name in df.columns
-            },
+            columns=prettify_header_name,
         )
         # Prettify index column name
         .rename_axis(prettify_header_name(df.index.name), axis="index")