ci: auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
chrislemke · Mar 12, 2024 · 18d3376 · 18d3376
1 parent b81daf3
commit 18d3376
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 16 deletions.
diff --git a/src/sk_transformers/generic_transformer.py b/src/sk_transformers/generic_transformer.py
@@ -291,9 +291,11 @@ def transform(self, X: pd.DataFrame) -> pd.DataFrame:
 
             agg_df = X.groupby(groupby_columns).agg(
                 [
-                    getattr(pl, agg_func)(agg_column).alias(agg_new_column)
-                    if isinstance(agg_func, str)
-                    else pl.col(agg_column).apply(agg_func).alias(agg_new_column)
+                    (
+                        getattr(pl, agg_func)(agg_column).alias(agg_new_column)
+                        if isinstance(agg_func, str)
+                        else pl.col(agg_column).apply(agg_func).alias(agg_new_column)
+                    )
                     for (agg_column, agg_func, agg_new_column) in agg_features
                 ]
             )

diff --git a/src/sk_transformers/string_transformer.py b/src/sk_transformers/string_transformer.py
@@ -489,9 +489,11 @@ def __init__(
     ) -> None:
         super().__init__()
         self.features = [
-            (split_tuple[0], split_tuple[1], split_tuple[2])
-            if len(split_tuple) == 3
-            else (split_tuple[0], split_tuple[1], -1)
+            (
+                (split_tuple[0], split_tuple[1], split_tuple[2])
+                if len(split_tuple) == 3
+                else (split_tuple[0], split_tuple[1], -1)
+            )
             for split_tuple in features
         ]
 
@@ -515,20 +517,22 @@ def transform(self, X: pd.DataFrame) -> pd.DataFrame:
         ]
 
         select_with_expr = [
-            pl.col(column)
-            .str.splitn(by=separator, n=max_possible_split + 1)
-            .struct.rename_fields(
-                [column + f"_part_{i}" for i in range(1, max_possible_split + 2)]
-            )
-            .alias(column + "_alias")
-            if maxsplit in [0, -1] or maxsplit > max_possible_split
-            else (
+            (
                 pl.col(column)
-                .str.splitn(by=separator, n=maxsplit + 1)
+                .str.splitn(by=separator, n=max_possible_split + 1)
                 .struct.rename_fields(
-                    [column + f"_part_{i}" for i in range(1, maxsplit + 2)]
+                    [column + f"_part_{i}" for i in range(1, max_possible_split + 2)]
                 )
                 .alias(column + "_alias")
+                if maxsplit in [0, -1] or maxsplit > max_possible_split
+                else (
+                    pl.col(column)
+                    .str.splitn(by=separator, n=maxsplit + 1)
+                    .struct.rename_fields(
+                        [column + f"_part_{i}" for i in range(1, maxsplit + 2)]
+                    )
+                    .alias(column + "_alias")
+                )
             )
             for (column, separator, maxsplit), max_possible_split in zip(
                 self.features, max_possible_splits_list