precommit

aphp · Apr 17, 2024 · 9c91724 · 9c91724
1 parent 3569ca3
commit 9c91724
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/eds_scikit/utils/sort_first_koalas.py b/eds_scikit/utils/sort_first_koalas.py
@@ -1,7 +1,11 @@
-from eds_scikit.utils.typing import DataFrame
 from typing import List
 
-def sort_values_first_koalas(dataframe : DataFrame , by_cols : List[str], cols : List[str], ascending : bool = True) -> DataFrame:
+from eds_scikit.utils.typing import DataFrame
+
+
+def sort_values_first_koalas(
+    dataframe: DataFrame, by_cols: List[str], cols: List[str], ascending: bool = True
+) -> DataFrame:
     """Replacement for dataframe.sort_values(cols).groupby(by_cols).first()
 
     To get a deterministic first, provide an id column of your dataframe as the last element of cols.
@@ -17,7 +21,7 @@ def sort_values_first_koalas(dataframe : DataFrame , by_cols : List[str], cols :
     -------
     DataFrame
 
-        """
+    """
     for col in cols:
         dataframe_min_max = dataframe.groupby(by_cols, as_index=False)[col]
         dataframe_min_max = (