Simplify column rename when outputting resulting dataframe

caleb531 · Feb 22, 2024 · 20e1686 · 20e1686
1 parent 7b689c3
commit 20e1686
Showing 1 changed file with 1 addition and 4 deletions.
diff --git 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")