Refactor _get_column_expr and opt_dtype for improved readability by formatting function arguments

Author: legout

src/flowerpower/plugins/io/helpers/polars.py

@@ -133,7 +133,11 @@ def _optimize_string_column(
 
 
 def _get_column_expr(
-    df: pl.DataFrame, col_name: str, shrink_numerics: bool, allow_unsigned: bool, time_zone: str | None = None
+    df: pl.DataFrame,
+    col_name: str,
+    shrink_numerics: bool,
+    allow_unsigned: bool,
+    time_zone: str | None = None,
 ) -> pl.Expr:
     """Generate optimization expression for a single column."""
     series = df[col_name]
@@ -144,7 +148,9 @@ def _get_column_expr(
 
     # Process based on current type
     if series.dtype.is_numeric():
-        return _optimize_numeric_column(series, col_name, shrink_numerics, allow_unsigned)
+        return _optimize_numeric_column(
+            series, col_name, shrink_numerics, allow_unsigned
+        )
     elif series.dtype == pl.Utf8:
         return _optimize_string_column(series, col_name, shrink_numerics, time_zone)
 
@@ -198,7 +204,9 @@ def opt_dtype(
     for col_name in cols_to_process:
         try:
             expressions.append(
-                _get_column_expr(df, col_name, shrink_numerics, allow_unsigned, time_zone)
+                _get_column_expr(
+                    df, col_name, shrink_numerics, allow_unsigned, time_zone
+                )
             )
         except Exception as e:
             if strict: